Practical Uses of SymPy for Symbolic Mathematics in Python

Practical Uses of SymPy for Symbolic Mathematics in Python

$SymPy$ is a powerful $Python$ library for symbolic mathematics, which allows you to perform algebraic manipulations, calculus, equation solving, and more, all symbolically.

Here are some practical and convenient ways to use $SymPy$:

1. Symbolic Algebra:

  • Defining Symbols:
    You can define symbols for algebraic expressions using symbols() or Symbol(). 
    1
    2
    3
    4
    5
    from sympy import symbols

    x, y = symbols('x y')
    expression = x + 2*y
    print(expression)
  • Output: 
    1
    x + 2*y
    This allows you to perform operations like simplification, expansion, and factorization symbolically.
  • Simplification: 
    1
    2
    3
    4
    5
    from sympy import simplify

    expr = (x**2 + 2*x + 1)/(x + 1)
    simplified_expr = simplify(expr)
    print(simplified_expr)
  • Output: 
    1
    x + 1

2. Solving Equations:

  • Solving Algebraic Equations:
    $SymPy$ can solve equations for you symbolically. 
    1
    2
    3
    4
    5
    from sympy import Eq, solve

    equation = Eq(x**2 - 4, 0)
    solutions = solve(equation, x)
    print(solutions)
  • Output: 
    1
    [-2, 2]
  • Solving Systems of Equations:
    $SymPy$ also handles systems of linear or nonlinear equations. 
    1
    2
    3
    4
    equation1 = Eq(x + y, 2)
    equation2 = Eq(x - y, 0)
    solutions = solve((equation1, equation2), (x, y))
    print(solutions)
  • Output: 
    1
    {x: 1, y: 1}

3. Calculus:

  • Differentiation:
    Perform symbolic differentiation with ease. 
    1
    2
    3
    4
    5
    from sympy import diff

    function = x**3 + 3*x**2 + 2*x + 1
    derivative = diff(function, x)
    print(derivative)
  • Output: 
    1
    3*x**2 + 6*x + 2
  • Integration:
    Symbolic integration (both definite and indefinite) is straightforward. 
    1
    2
    3
    4
    5
    6
    7
    8
    from sympy import integrate

    integral = integrate(function, x)
    print(integral)

    # Definite integral
    definite_integral = integrate(function, (x, 0, 2))
    print(definite_integral)
  • Output: 
    1
    2
    x**4/4 + x**3 + x**2 + x
    18

4. Series Expansion:

  • Taylor Series:
    You can find the Taylor series expansion of functions. 
    1
    2
    3
    4
    from sympy import symbols, sin, series

    expansion = series(sin(x), x, 0, 6)
    print(expansion)
  • Output: 
    1
    x - x**3/6 + x**5/120 + O(x**6)

5. Matrix Operations:

  • Symbolic Matrices:
    $SymPy$ can handle symbolic matrices and perform operations like inversion, determinant calculation, and eigenvalue finding. 
    1
    2
    3
    4
    5
    from sympy import Matrix

    matrix = Matrix([[x, 2], [3, y]])
    determinant = matrix.det()
    print(determinant)
  • Output: 
    1
    x*y - 6

6. Limits:

  • Calculating Limits:
    $SymPy$ allows you to calculate limits of functions. 
    1
    2
    3
    4
    5
    from sympy import limit

    function = (x**2 - 1)/(x - 1)
    lim = limit(function, x, 1)
    print(lim)
  • Output: 
    1
    2

7. Solving Differential Equations:

  • Ordinary Differential Equations:
    $SymPy$ can symbolically solve ODEs. 
    1
    2
    3
    4
    5
    6
    from sympy import Function, dsolve, Eq

    f = Function('f')
    ode = Eq(f(x).diff(x, x) - 2*f(x), 0)
    solution = dsolve(ode, f(x))
    print(solution)
  • Output: 
    1
    Eq(f(x), C1*exp(-sqrt(2)*x) + C2*exp(sqrt(2)*x))

8. Plotting:

  • Basic Plotting:
    $SymPy$ integrates with matplotlib for basic plotting. 
    1
    2
    3
    from sympy.plotting import plot

    plot(x**2 + 2*x + 1)
  • Output:

9. Substitution:

  • Substituting Values:
    Substitute specific values into expressions. 
    1
    2
    3
    expr = x**2 + 2*x + 1
    result = expr.subs(x, 2)
    print(result)
  • Output: 
    1
    9

10. Polynomial Manipulation:

  • Factoring and Expanding Polynomials:
    $SymPy$ can factor and expand polynomial expressions. 
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    from sympy import symbols, factor, expand

    x = symbols('x')

    # Factor the polynomial x^2 - 4
    factored = factor(x**2 - 4)

    # Expand the expression (x + 2)(x - 2)
    expanded = expand((x + 2)*(x - 2))

    print(factored)
    print(expanded)
  • Output: 
    1
    2
    (x - 2)*(x + 2)
    x**2 - 4

These examples showcase some of the most common and convenient uses of $SymPy$.

Whether you’re working on algebra, calculus, or solving equations, $SymPy$ provides a powerful toolset for symbolic mathematics in $Python$.

Knapsack Problem

Example of a Knapsack Problem

Problem Statement:
You are given a set of items, each with a weight and a value.

You need to determine the number of each item to include in a knapsack so that the total weight is less than or equal to a given limit, and the total value is as large as possible.

Items:

  • Item 1: Weight = $2$, Value = $3$
  • Item 2: Weight = $3$, Value = $4$
  • Item 3: Weight = $4$, Value = $5$
  • Item 4: Weight = $5$, Value = $6$

Knapsack Capacity: $8$

The goal is to maximize the total value without exceeding the knapsack capacity of $8$.

Python Solution:

This is a $dynamic$ $programming$ approach to solve the $0$/$1$ Knapsack Problem.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
def knapsack(values, weights, capacity):
    n = len(values)
    # Create a 2D DP table with (n+1) x (capacity+1) size
    dp = [[0 for _ in range(capacity + 1)] for _ in range(n + 1)]

    # Fill the DP table
    for i in range(1, n + 1):
        for w in range(1, capacity + 1):
            if weights[i - 1] <= w:
                dp[i][w] = max(dp[i - 1][w], dp[i - 1][w - weights[i - 1]] + values[i - 1])
            else:
                dp[i][w] = dp[i - 1][w]

    # The maximum value will be in dp[n][capacity]
    return dp[n][capacity]

# Example data
values = [3, 4, 5, 6]
weights = [2, 3, 4, 5]
capacity = 8

# Solve the problem
max_value = knapsack(values, weights, capacity)
print("Maximum value that can be carried in the knapsack:", max_value)

Explanation:

  1. Initialization:

    • The function knapsack takes three arguments: values, weights, and capacity.
    • The $dynamic$ $programming$ ($DP$) table dp is initialized with zeros.
      The table has dimensions (n+1) x (capacity+1), where n is the number of items.

  2. Filling the DP Table:

    • The outer loop iterates through the items, and the inner loop iterates through possible capacities from 1 to capacity.
    • For each item and capacity, the $DP$ table is updated by considering two possibilities:
      • Not including the item in the knapsack (value remains the same as the previous row).
      • Including the item in the knapsack (value is updated based on the remaining capacity and the value of the item).

  3. Result:

    • After filling the $DP$ table, the maximum value that can be carried in the knapsack is found in dp[n][capacity].

Output:

1
Maximum value that can be carried in the knapsack: 10

In this example, the optimal solution is to include Item $2$ (weight $3$, value $4$) and Item $4$ (weight $5$, value $6$), which gives a total value of $10$ while staying within the capacity limit of $8$.

Scheduling Problem

Scheduling Problem

A more practical application of $python$-$constraint$ is solving $scheduling$ $problems$, such as assigning people to tasks without conflicts.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from constraint import Problem

problem = Problem()

# Add variables (people and time slots)
people = ["Alice", "Bob", "Charlie"]
slots = ["Slot1", "Slot2", "Slot3"]
problem.addVariables(people, slots)

# Add constraints: no two people can be in the same time slot
problem.addConstraint(lambda a, b: a != b, ("Alice", "Bob"))
problem.addConstraint(lambda a, c: a != c, ("Alice", "Charlie"))
problem.addConstraint(lambda b, c: b != c, ("Bob", "Charlie"))

# Get the solution
solution = problem.getSolution()

# Print the solution
print(solution)

Explanation

This $Python$ code uses the $python$-$constraint$ library to solve a scheduling problem where three people (Alice, Bob, and Charlie) need to be assigned to different time slots (Slot1, Slot2, and Slot3).

The goal is to ensure that no two people are assigned to the same time slot.

Step-by-Step Explanation:

  1. Importing the Required Module:

    1
    from constraint import Problem

    The Problem class from the $python$-$constraint$ library is used to define and solve the constraint satisfaction problem (CSP).

  2. Creating the Problem Instance:

    1
    problem = Problem()

    A Problem object is instantiated, which will be used to define the variables (people and time slots) and constraints (no two people in the same slot).

  3. Defining Variables:

    1
    2
    3
    people = ["Alice", "Bob", "Charlie"]
    slots = ["Slot1", "Slot2", "Slot3"]
    problem.addVariables(people, slots)

    Three variables (Alice, Bob, and Charlie) are created, representing the people who need to be assigned to time slots.
    Each variable can take one of the values from the slots list (Slot1, Slot2, Slot3).

  4. Adding Constraints:

    1
    2
    3
    problem.addConstraint(lambda a, b: a != b, ("Alice", "Bob"))
    problem.addConstraint(lambda a, c: a != c, ("Alice", "Charlie"))
    problem.addConstraint(lambda b, c: b != c, ("Bob", "Charlie"))

    These constraints ensure that no two people are assigned to the same time slot:

    • The first constraint ensures Alice and Bob do not share the same slot.
    • The second constraint ensures Alice and Charlie do not share the same slot.
    • The third constraint ensures Bob and Charlie do not share the same slot.

    These constraints effectively enforce that each person must be in a unique time slot.

  5. Solving the Problem:

    1
    solution = problem.getSolution()

    The getSolution() method is used to find a solution that satisfies all the constraints.
    The solution is returned as a dictionary where the keys are the names of the people and the values are the assigned time slots.

  6. Printing the Solution:

    1
    print(solution)

    The solution is printed, showing which time slot each person is assigned to.

    For example, the output might be something like {'Alice': 'Slot3', 'Bob': 'Slot2', 'Charlie': 'Slot1'}, indicating that Alice is in Slot3, Bob is in Slot2, and Charlie is in Slot1.

Summary

This code solves a simple scheduling problem using constraints to ensure that three people (Alice, Bob, and Charlie) are assigned to different time slots (Slot1, Slot2, and Slot3).

The constraints ensure that no two people share the same time slot, and the solution provides a valid assignment that meets these conditions.

Constructing a $3 \times 3$ Magic Square

Constructing a $3 \times 3$ Magic Square

The $python$-$constraint$ library is a powerful tool for solving constraint satisfaction problems (CSPs) in $Python$.

It allows you to define variables, domains, and constraints, then finds solutions that satisfy all the constraints.

While the basic usage is straightforward, here are some unconventional or “curious” ways to use $python$-$constraint$ that showcase its flexibility.

Magic Square

A $magic$ $square$ is a grid where the sum of the numbers in each row, column, and diagonal is the same.

This is another interesting constraint satisfaction problem that can be solved with $python$-$constraint$.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from constraint import Problem, AllDifferentConstraint

problem = Problem()

# Add variables for a 3x3 magic square
variables = [(row, col) for row in range(3) for col in range(3)]
problem.addVariables(variables, range(1, 10))

# Add the all-different constraint
problem.addConstraint(AllDifferentConstraint(), variables)

# Add the magic square sum constraints
magic_sum = 15
for row in range(3):
    problem.addConstraint(lambda *args: sum(args) == magic_sum, [(row, col) for col in range(3)])

for col in range(3):
    problem.addConstraint(lambda *args: sum(args) == magic_sum, [(row, col) for row in range(3)])

problem.addConstraint(lambda *args: sum(args) == magic_sum, [(i, i) for i in range(3)])
problem.addConstraint(lambda *args: sum(args) == magic_sum, [(i, 2 - i) for i in range(3)])

# Get the solution
solution = problem.getSolution()

# Print the solution in a grid format
if solution:
    for row in range(3):
        print([solution[(row, col)] for col in range(3)])
else:
    print("No solution found.")

Explanation

This $Python$ code uses the $python$-$constraint$ library to solve a constraint satisfaction problem, specifically constructing a $ 3 \times 3 $ magic square.

In a magic square, all rows, columns, and diagonals must sum to the same value, known as the “$magic$ $sum$.”

For a $3 \times 3$ grid with the numbers $1$ to $9$, this magic sum is $15$.

Step-by-Step Explanation:

  1. Importing Required Modules:

    1
    from constraint import Problem, AllDifferentConstraint

    The Problem class is used to define and solve the constraint satisfaction problem (CSP).
    The AllDifferentConstraint ensures that all variables (the cells in the magic square) take different values.

  2. Creating the Problem Instance:

    1
    problem = Problem()

    A Problem object is instantiated to manage the variables and constraints of the problem.

  3. Defining Variables:

    1
    2
    variables = [(row, col) for row in range(3) for col in range(3)]
    problem.addVariables(variables, range(1, 10))

    The variables represent each cell in the $3 \times 3$ grid, identified by their row and column indices (e.g., (0, 0) for the top-left cell).
    Each cell can take a value from $1$ to $9$.

  4. Adding the All-Different Constraint:

    1
    problem.addConstraint(AllDifferentConstraint(), variables)

    This constraint ensures that all cells in the magic square have different values.
    In other words, no two cells can have the same number.

  5. Defining the Magic Sum:

    1
    magic_sum = 15

    The sum of each row, column, and diagonal in the magic square must equal $15$, which is the “$magic$ $sum$” for a $3 \times 3$ magic square using numbers $1$ to $9$.

  6. Adding Row Constraints:

    1
    2
    for row in range(3):
        problem.addConstraint(lambda *args: sum(args) == magic_sum, [(row, col) for col in range(3)])

    This loop adds constraints for each row in the grid, ensuring that the sum of the numbers in each row equals $15$.

  7. Adding Column Constraints:

    1
    2
    for col in range(3):
        problem.addConstraint(lambda *args: sum(args) == magic_sum, [(row, col) for row in range(3)])

    This loop adds constraints for each column in the grid, ensuring that the sum of the numbers in each column equals $15$.

  8. Adding Diagonal Constraints:

    1
    2
    problem.addConstraint(lambda *args: sum(args) == magic_sum, [(i, i) for i in range(3)])
    problem.addConstraint(lambda *args: sum(args) == magic_sum, [(i, 2 - i) for i in range(3)])

    These lines add constraints for the two diagonals of the magic square.
    The first constraint ensures that the main diagonal (top-left to bottom-right) sums to $15$, and the second ensures that the anti-diagonal (top-right to bottom-left) sums to $15$.

  9. Solving the Problem:

    1
    solution = problem.getSolution()

    The getSolution() method attempts to find a solution that satisfies all the constraints.
    If a solution is found, it is returned as a dictionary, where each key is a cell (e.g., (0, 0)) and each value is the number assigned to that cell.

  10. Printing the Solution:

    1
    2
    3
    4
    5
    if solution:
        for row in range(3):
            print([solution[(row, col)] for col in range(3)])
    else:
        print("No solution found.")

    If a solution is found, the code prints the magic square in a $3 \times 3$ grid format.
    If no solution is found, it prints “No solution found.”

Summary:

This code constructs a $3 \times 3$ magic square where each number from $1$ to $9$ is used exactly once, and the sum of the numbers in each row, column, and diagonal equals $15$.

By defining the problem as a set of constraints and solving it with the python-constraint library, the code finds a valid configuration for the magic square.

Output

[Output]

1
2
3
[6, 1, 8]
[7, 5, 3]
[2, 9, 4]

The result displayed represents a $3 \times 3$ magic square.

A magic square is a grid in which the numbers in each row, column, and diagonal all add up to the same value, known as the “$magic$ $sum$.”

For a $3 \times 3$ magic square using the numbers $1$ to $9$, this sum is $15$.

Breakdown of the Result:

  • First Row: [6, 1, 8]
    Sum: 6 + 1 + 8 = 15

  • Second Row: [7, 5, 3]
    Sum: 7 + 5 + 3 = 15

  • Third Row: [2, 9, 4]
    Sum: 2 + 9 + 4 = 15

Column Sums:

  • First Column: 6 + 7 + 2 = 15
  • Second Column: 1 + 5 + 9 = 15
  • Third Column: 8 + 3 + 4 = 15

Diagonal Sums:

  • Main Diagonal (Top-Left to Bottom-Right): 6 + 5 + 4 = 15
  • Anti-Diagonal (Top-Right to Bottom-Left): 8 + 5 + 2 = 15

Conclusion:

This $3 \times 3$ grid is a valid magic square.

Every row, column, and diagonal sums to $15$, satisfying the conditions of the magic square.

The numbers $1$ to $9$ are used exactly once, and the constraints of the problem have been correctly applied to generate this solution.

Cryptarithmetic Puzzle with Python-Constraint

Cryptarithmetic Puzzlewith Python-Constraint: A Creative Approach to Constraint Satisfaction

The $python$-$constraint$ library is a powerful tool for solving constraint satisfaction problems (CSPs) in $Python$.

It allows you to define variables, domains, and constraints, then finds solutions that satisfy all the constraints.

While the basic usage is straightforward, here are some unconventional or “curious” ways to use $python$-$constraint$ that showcase its flexibility.

Cryptarithmetic Puzzle

$Cryptarithmetic$ $puzzles$ involve solving mathematical equations where digits are replaced with letters.

For example, in the puzzle SEND + MORE = MONEY, each letter represents a unique digit.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from constraint import Problem, AllDifferentConstraint

problem = Problem()

# Add variables for each letter
letters = 'SENDMORY'
problem.addVariables(letters, range(10))

# Add constraints: all letters must represent different digits
problem.addConstraint(AllDifferentConstraint(), letters)

# Add leading digit constraints (e.g., S and M cannot be zero)
problem.addConstraint(lambda S: S != 0, 'S')
problem.addConstraint(lambda M: M != 0, 'M')

# Add the equation constraint: SEND + MORE = MONEY
def equation(S, E, N, D, M, O, R, Y):
    send = S * 1000 + E * 100 + N * 10 + D
    more = M * 1000 + O * 100 + R * 10 + E
    money = M * 10000 + O * 1000 + N * 100 + E * 10 + Y
    return send + more == money

problem.addConstraint(equation, 'SENDMORY')

# Get the solution
solution = problem.getSolution()

# Print the solution
if solution:
    print(solution)
else:
    print("No solution found.")

Explanation

This $Python$ code uses the python-constraint library to solve a $cryptarithmetic$ $puzzle$, specifically the puzzle SEND + MORE = MONEY.

In this type of puzzle, each letter represents a unique digit, and the goal is to find the digits that satisfy the equation.

Step-by-Step Explanation:

  1. Importing the Required Modules:

    1
    from constraint import Problem, AllDifferentConstraint

    The Problem class from the python-constraint library is used to define and solve constraint satisfaction problems (CSPs).
    The AllDifferentConstraint ensures that all variables (letters) represent different digits.

  2. Creating the Problem Instance:

    1
    problem = Problem()

    A Problem object is instantiated, which will be used to define the variables and constraints for this puzzle.

  3. Defining Variables:

    1
    2
    letters = 'SENDMORY'
    problem.addVariables(letters, range(10))

    The letters in the puzzle (‘S’, ‘E’, ‘N’, ‘D’, ‘M’, ‘O’, ‘R’, ‘Y’) are treated as variables.
    Each letter can take a value between $0$ and $9$, representing a digit.

  4. Adding the All-Different Constraint:

    1
    problem.addConstraint(AllDifferentConstraint(), letters)

    This constraint ensures that all letters represent different digits.
    For example, ‘S’ cannot be the same as ‘E’, and so on.

  5. Adding Leading Digit Constraints:

    1
    2
    problem.addConstraint(lambda S: S != 0, 'S')
    problem.addConstraint(lambda M: M != 0, 'M')

    These constraints ensure that the leading digits of the numbers formed by the letters cannot be zero.
    For example, ‘S’ (the first digit of “SEND”) and ‘M’ (the first digit of “MORE” and “MONEY”) must be non-zero.

  6. Defining the Equation Constraint:

    1
    2
    3
    4
    5
    def equation(S, E, N, D, M, O, R, Y):
        send = S * 1000 + E * 100 + N * 10 + D
        more = M * 1000 + O * 100 + R * 10 + E
        money = M * 10000 + O * 1000 + N * 100 + E * 10 + Y
        return send + more == money

    This function converts the letters into the corresponding numbers and defines the equation SEND + MORE = MONEY.
    It calculates the numeric values of “SEND”, “MORE”, and “MONEY” and checks if the sum of “SEND” and “MORE” equals “MONEY”.

  7. Adding the Equation Constraint to the Problem:

    1
    problem.addConstraint(equation, 'SENDMORY')

    The equation function is added as a constraint to the problem, ensuring that the solution must satisfy the equation SEND + MORE = MONEY.

  8. Solving the Problem:

    1
    solution = problem.getSolution()

    The getSolution() method attempts to find a solution that satisfies all the constraints.
    If a solution exists, it returns the solution as a dictionary mapping each letter to its corresponding digit.

  9. Printing the Solution:

    1
    2
    3
    4
    if solution:
        print(solution)
    else:
        print("No solution found.")

    If a solution is found, it is printed. Otherwise, the program outputs “No solution found.”

Summary:

This code solves the $cryptarithmetic$ $puzzle$ SEND + MORE = MONEY using constraint satisfaction techniques.

It assigns digits to each letter while ensuring that the digits satisfy the puzzle’s rules:
no two letters have the same digit, the first letters (‘S’ and ‘M’) are non-zero, and the equation SEND + MORE = MONEY holds true.

Output

[Output]

1
{'M': 1, 'S': 9, 'D': 7, 'E': 5, 'N': 6, 'O': 0, 'R': 8, 'Y': 2}

The result {'M': 1, 'S': 9, 'D': 7, 'E': 5, 'N': 6, 'O': 0, 'R': 8, 'Y': 2} represents a valid solution to the $cryptarithmetic$ $puzzle$ SEND + MORE = MONEY.

Explanation of the Solution:

  • Letter-to-Digit Mapping:
    • M = 1
    • S = 9
    • D = 7
    • E = 5
    • N = 6
    • O = 0
    • R = 8
    • Y = 2

Substitution in the Equation:

  • SEND becomes 9567 (S = 9, E = 5, N = 6, D = 7)
  • MORE becomes 1085 (M = 1, O = 0, R = 8, E = 5)
  • MONEY becomes 10652 (M = 1, O = 0, N = 6, E = 5, Y = 2)

Verification:

The equation SEND + MORE = MONEY can be verified as follows:

  • 9567 + 1085 = 10652
  • The sum is correct, and the equation holds true.

Conclusion:

This mapping of letters to digits correctly satisfies the $cryptarithmetic$ $puzzle$, making it a valid solution.

The unique digits assigned to each letter, along with the correct calculation, ensure that the equation SEND + MORE = MONEY is fulfilled.

Solving Sudoku with Python-Constraint

Solving Sudoku with Python-Constraint: A Creative Approach to Constraint Satisfaction

The $python-constraint$ library is a powerful tool for solving constraint satisfaction problems (CSPs) in Python.

It allows you to define variables, domains, and constraints, then finds solutions that satisfy all the constraints.

While the basic usage is straightforward, here are some unconventional or “curious” ways to use $python$-$constraint$ that showcase its flexibility.

Sudoku Solver

You can use $python$-$constraint$ to solve a $Sudoku$ $puzzle$, which is a fun and challenging constraint satisfaction problem.

Here, each cell in the $Sudoku$ grid is treated as a variable, and the constraints ensure that each row, column, and $3 \times 3$ sub-grid contains unique numbers.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from constraint import Problem, AllDifferentConstraint

# Initialize the problem
problem = Problem()

# Add variables (81 variables for a 9x9 Sudoku grid)
for row in range(9):
    for col in range(9):
        problem.addVariable((row, col), range(1, 10))

# Add row constraints
for row in range(9):
    problem.addConstraint(AllDifferentConstraint(), [(row, col) for col in range(9)])

# Add column constraints
for col in range(9):
    problem.addConstraint(AllDifferentConstraint(), [(row, col) for row in range(9)])

# Add 3x3 sub-grid constraints
for row in range(0, 9, 3):
    for col in range(0, 9, 3):
        cells = [(row + r, col + c) for r in range(3) for c in range(3)]
        problem.addConstraint(AllDifferentConstraint(), cells)

# Optionally, set some initial values (partial Sudoku grid)
initial_values = {(0, 1): 3, (0, 2): 6, (1, 0): 9, (1, 3): 5, (1, 5): 7, (2, 1): 1}
for (row, col), value in initial_values.items():
    problem.addConstraint(lambda var, val=value: var == val, [(row, col)])

# Get the solution
solution = problem.getSolution()

# Print the solution in a grid format
if solution:
    for row in range(9):
        print([solution[(row, col)] for col in range(9)])
else:
    print("No solution found.")

Output

[Output]

1
2
3
4
5
6
7
8
9
[7, 3, 6, 4, 2, 1, 9, 8, 5]
[9, 8, 4, 5, 6, 7, 3, 2, 1]
[5, 1, 2, 9, 8, 3, 7, 6, 4]
[8, 5, 9, 3, 4, 6, 2, 1, 7]
[2, 6, 7, 8, 1, 9, 5, 4, 3]
[1, 4, 3, 7, 5, 2, 8, 9, 6]
[6, 9, 5, 2, 7, 4, 1, 3, 8]
[4, 2, 8, 1, 3, 5, 6, 7, 9]
[3, 7, 1, 6, 9, 8, 4, 5, 2]

The result is a solution to a $Sudoku$ $puzzle$, represented as a $9 \times 9$ grid where each row is listed in sequence.

Let’s analyze the result in the context of $Sudoku$ rules.

Sudoku Rules Recap:

  1. Rows:
    Each row must contain the digits $1$ to $9$ without repetition.
  2. Columns:
    Each column must contain the digits $1$ to $9$ without repetition.
  3. 3x3 Sub-Grids:
    Each of the nine $3 \times 3$ sub-grids must contain the digits $1$ to $9$ without repetition.

Explanation of the Result:

The grid you provided follows the standard format of a $Sudoku$ $puzzle$ solution.

Each list corresponds to a row in the grid, from top to bottom.

  • Row 1: [7, 3, 6, 4, 2, 1, 9, 8, 5]
  • Row 2: [9, 8, 4, 5, 6, 7, 3, 2, 1]
  • Row 3: [5, 1, 2, 9, 8, 3, 7, 6, 4]
  • Row 4: [8, 5, 9, 3, 4, 6, 2, 1, 7]
  • Row 5: [2, 6, 7, 8, 1, 9, 5, 4, 3]
  • Row 6: [1, 4, 3, 7, 5, 2, 8, 9, 6]
  • Row 7: [6, 9, 5, 2, 7, 4, 1, 3, 8]
  • Row 8: [4, 2, 8, 1, 3, 5, 6, 7, 9]
  • Row 9: [3, 7, 1, 6, 9, 8, 4, 5, 2]

Validation:

  • Rows:
    Each row contains the digits $1$ to $9$ without any duplicates.
  • Columns:
    By inspecting each vertical slice (not shown explicitly), each column also contains the digits $1$ to $9$ without any duplicates.
  • Sub-Grids:
    The $3 \times 3$ sub-grids (e.g., top-left $3 \times 3$, top-middle $3 \times 3$, etc.) contain all digits from $1$ to $9$ without repetition.

Conclusion:

This grid appears to be a valid solution to the $Sudoku$ $puzzle$.

It satisfies all the constraints of the game, ensuring that each row, column, and sub-grid contains the digits $1$ to $9$ without any repetition.

Creating Interactive and Customizable Visualizations with Altair

Creating Interactive and Customizable Visualizations with Altair

$Altair$ is a declarative statistical visualization library for $Python$, which is built on top of the powerful Vega and Vega-Lite visualization grammars.

It enables you to create complex visualizations with concise, readable code.

Here’s an example of how to use $Altair$ to create different types of charts.

1. Basic Scatter Plot

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
!pip install altair_viewer
import altair as alt
import pandas as pd

# Sample data
data = pd.DataFrame({
    'X': [1, 2, 3, 4, 5],
    'Y': [2, 3, 5, 7, 11],
    'Category': ['A', 'B', 'A', 'B', 'A']
})

# Scatter plot
scatter_plot = alt.Chart(data).mark_point().encode(
    x='X',
    y='Y',
    color='Category'
)

scatter_plot.show()

[Output]

2. Line Chart

1
2
3
4
5
6
7
8
# Line chart
line_chart = alt.Chart(data).mark_line().encode(
    x='X',
    y='Y',
    color='Category'
)

line_chart.show()

[Output]

3. Bar Chart

1
2
3
4
5
6
7
# Bar chart
bar_chart = alt.Chart(data).mark_bar().encode(
    x='Category',
    y='sum(Y)',
)

bar_chart.show()

[Output]

4. Interactive Visualization

Altair supports interactive visualizations. Below is an example of a chart with a tooltip.

1
2
3
4
5
6
7
8
9
# Interactive scatter plot with tooltips
interactive_scatter = alt.Chart(data).mark_point().encode(
    x='X',
    y='Y',
    color='Category',
    tooltip=['X', 'Y', 'Category']
).interactive()

interactive_scatter.show()

[Output]

5. Faceted Charts

Faceting allows you to create small multiples of a plot based on a categorical variable.

1
2
3
4
5
6
7
8
9
10
# Faceted scatter plot
faceted_chart = alt.Chart(data).mark_point().encode(
    x='X',
    y='Y',
    color='Category'
).facet(
    column='Category'
)

faceted_chart.show()

[Output]

6. Layered Charts

You can layer multiple charts on top of each other.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# Layered chart
points = alt.Chart(data).mark_point().encode(
    x='X',
    y='Y',
    color='Category'
)

lines = alt.Chart(data).mark_line().encode(
    x='X',
    y='Y'
)

layered_chart = points + lines

layered_chart.show()

[Output]

7. Customizing Appearance

You can customize various aspects of the chart, like axis labels, titles, and themes.

1
2
3
4
5
6
7
8
9
10
# Customized chart
custom_chart = alt.Chart(data).mark_point().encode(
    x=alt.X('X', axis=alt.Axis(title='Custom X-axis Label')),
    y=alt.Y('Y', axis=alt.Axis(title='Custom Y-axis Label')),
    color='Category'
).properties(
    title='Customized Chart'
)

custom_chart.show()

[Output]

Conclusion

$Altair$ provides a straightforward, expressive, and powerful way to create visualizations in $Python$.

With its declarative syntax, you can focus on the data and the visual representation rather than low-level details of rendering.

Optimizing Vaccine Distribution to Minimize Infections

Optimizing Vaccine Distribution to Minimize Infections

Let’s solve an $optimization$ $problem$ related to $healthcare$, specifically in managing the allocation of limited medical resources (e.g., vaccines, medications, or hospital beds) to minimize the number of people affected by a disease.

Problem Statement:

Suppose you are managing a limited supply of vaccines that can be distributed across different regions to minimize the spread of a disease.

The goal is to optimize the allocation of these vaccines so that the total number of infected individuals is minimized.

Assumptions:

  1. There are n regions, each with a certain population and a number of infected individuals.
  2. The effectiveness of the vaccine in reducing the number of infections is proportional to the number of vaccines allocated to each region.
  3. The total number of vaccines is limited.

Formulation:

  • $( x_i )$ be the number of vaccines allocated to region $( i )$,
  • $( p_i )$ be the population of region $( i )$,
  • $( r_i )$ be the current infection rate in region $( i )$,
  • $( v )$ be the total number of vaccines available.

The objective is to minimize the total number of infected people across all regions after vaccine distribution.

$$
\text{Minimize } \sum_{i=1}^n \left( p_i \cdot r_i - \alpha \cdot x_i \right)
$$

subject to the constraint:

$$
\sum_{i=1}^n x_i = v \quad \text{and} \quad x_i \geq 0 \text{ for all } i
$$

where $( \alpha )$ is a positive constant representing the effectiveness of the vaccine.

Python Code Using SciPy:

We can solve this optimization problem using Python’s SciPy library.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import numpy as np
from scipy.optimize import linprog

# Define the number of regions
n = 4  # For example, 4 regions

# Population in each region
population = np.array([1000, 1500, 2000, 2500])

# Infection rates in each region
infection_rate = np.array([0.05, 0.10, 0.07, 0.12])

# Vaccine effectiveness coefficient
alpha = 0.1

# Total number of vaccines available
total_vaccines = 1000

# Objective function coefficients (negative because linprog performs minimization)
c = -alpha * np.ones(n)

# Inequality constraints matrix and vector (no inequalities here, just an equality constraint)
A_eq = np.ones((1, n))  # Sum of all vaccines must equal total_vaccines
b_eq = np.array([total_vaccines])

# Bounds for each variable (number of vaccines allocated must be non-negative)
bounds = [(0, None)] * n

# Solve the linear programming problem
result = linprog(c, A_eq=A_eq, b_eq=b_eq, bounds=bounds, method='highs')

# Results
if result.success:
    print("Optimal vaccine distribution:", result.x)
    print("Total minimized infections:", np.dot(population, infection_rate) - np.dot(result.x, c))
else:
    print("Optimization failed.")

Explanation:

  1. Objective Function:

    • We minimize the total infections after vaccine allocation, modeled by $( -\alpha \cdot x_i )$, where $( x_i )$ is the number of vaccines allocated to region $( i )$.

  2. Constraints:

    • The sum of all vaccines distributed must equal the total available vaccines.
    • No region can receive a negative number of vaccines.

  3. SciPy Optimization:

    • We use the linprog() function from the SciPy library to solve this linear programming problem.

Results:

The output will show the optimal distribution of vaccines across the regions and the minimized total number of infections.

This method can be extended to more complex models, incorporating additional constraints or nonlinear relationships between variables.

Explanation of the Results:

  1. Optimal Vaccine Distribution: [1000. 0. 0. 0.]

    • The result indicates that all $1000$ available vaccines should be allocated entirely to the first region.
      This suggests that prioritizing the first region for vaccine distribution is the most effective way to minimize the total number of infections across all regions.

  2. Total Minimized Infections: 740.0

    • After distributing the vaccines according to the optimal strategy, the total number of infections across all regions has been minimized to $740$.
      This is the lowest possible number of infections that can be achieved given the available resources and the constraints of the problem.

Conclusion:

The optimization process has determined that focusing the entire vaccine supply on the first region will have the greatest impact in reducing the overall number of infections.

This outcome may be due to the specific infection rates and population sizes in each region.

Creating Various 3D Plots in Python with Matplotlib

Creating Various 3D Plots in Python with Matplotlib

Here are various examples of $3D$ $plots$ using $Python$’s $Matplotlib$ library.

These examples demonstrate how to create different types of 3D graphs, including surface plots, scatter plots, and wireframe plots.

Example 1: 3D Surface Plot

A $surface$ $plot$ is useful for visualizing a 3D surface.

Let’s plot the function:

$$
z = \sin(\sqrt{x^2 + y^2})
$$

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

# Create grid points
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
x, y = np.meshgrid(x, y)
z = np.sin(np.sqrt(x**2 + y**2))

# Create the 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the surface
surf = ax.plot_surface(x, y, z, cmap='viridis')

# Add color bar for reference
fig.colorbar(surf)

# Set labels
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title("3D Surface Plot")
plt.show()

Output:

Example 2: 3D Scatter Plot

A $3D$ $scatter$ $plot$ is useful for visualizing data points in three dimensions.

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# Generate random data
x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)

# Create the 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the scatter
ax.scatter(x, y, z, c=z, cmap='plasma')

# Set labels
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title("3D Scatter Plot")
plt.show()

Output:

Example 3: 3D Wireframe Plot

A $wireframe$ $plot$ shows the structure of a 3D surface using lines instead of a solid surface.

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Create grid points
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
x, y = np.meshgrid(x, y)
z = np.cos(np.sqrt(x**2 + y**2))

# Create the 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the wireframe
ax.plot_wireframe(x, y, z, color='blue')

# Set labels
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title("3D Wireframe Plot")
plt.show()

Output:

Example 4: 3D Contour Plot

A $contour$ $plot$ in 3D shows contour lines (constant values) of a 3D surface.

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Create grid points
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
x, y = np.meshgrid(x, y)
z = np.sin(x) * np.cos(y)

# Create the 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the contour
ax.contour3D(x, y, z, 50, cmap='coolwarm')

# Set labels
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title("3D Contour Plot")
plt.show()

Output:

Example 5: 3D Bar Plot

A $3D$ $bar$ $plot$ is useful for displaying data across three axes.

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# Data for bars
_x = np.arange(4)
_y = np.arange(3)
_xx, _yy = np.meshgrid(_x, _y)
x, y = _xx.ravel(), _yy.ravel()
z = np.zeros_like(x)

# Bar heights
dz = [1, 2, 3, 4, 2, 3, 1, 4, 2, 1, 3, 4]

# Create the 3D figure
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the 3D bars
ax.bar3d(x, y, z, 1, 1, dz, shade=True)

# Set labels
ax.set_xlabel('X axis')
ax.set_ylabel('Y axis')
ax.set_zlabel('Z axis')

plt.title("3D Bar Plot")
plt.show()

Output:

Conclusion

These examples show how to create various types of $3D$ $plots$ in $Python$ using $Matplotlib$.

You can customize these plots further by adjusting parameters like colors, axes labels, and grid points.

$3D$ $plotting$ is a powerful way to visualize complex data and mathematical functions in $Python$.

Mastering Symbolic Mathematics with SymPy

Mastering Symbolic Mathematics with SymPy

Here’s a sample code using $SymPy$, a $Python$ library for symbolic mathematics.

We’ll solve a quadratic equation symbolically and also demonstrate differentiation and integration.

Example 1: Solving a Quadratic Equation

Let’s solve the quadratic equation:

$$
x^2 - 5x + 6 = 0
$$

Python Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
import sympy as sp

# Define the symbol (variable)
x = sp.symbols('x')

# Define the quadratic equation
equation = x**2 - 5*x + 6

# Solve the equation
solutions = sp.solve(equation, x)

# Print the solutions
print("Solutions to the quadratic equation:")
print(solutions)

Explanation:

  1. Symbol Definition:

    • sp.symbols('x') creates a symbolic variable x.

  2. Equation:

    • We define the quadratic equation $ x^2 - 5x + 6 = 0 $.

  3. Solve:

    • sp.solve(equation, x) solves the equation for x and returns the roots (solutions).

  4. Output:

    • The roots of the quadratic equation are printed.

Result:

1
2
Solutions to the quadratic equation:
[2, 3]

Example 2: Differentiation

Let’s find the derivative of the function:

$$
f(x) = x^3 + 2x^2 - 3x + 1
$$

Python Code:

1
2
3
4
5
6
7
8
9
# Define the function
f = x**3 + 2*x**2 - 3*x + 1

# Compute the derivative
derivative = sp.diff(f, x)

# Print the derivative
print("The derivative of f(x) is:")
print(derivative)

Explanation:

  1. Function Definition:

    • We define the function $ f(x) = x^3 + 2x^2 - 3x + 1 $.

  2. Differentiation:

    • sp.diff(f, x) computes the derivative of f with respect to x.

  3. Output:

    • The derivative of the function is printed.

Result:

1
2
The derivative of f(x) is:
3*x**2 + 4*x - 3

Example 3: Integration

Let’s compute the indefinite integral of the function:

$$
f(x) = 3x^2 - 4x + 5
$$

Python Code:

1
2
3
4
5
6
7
8
9
# Define the function
g = 3*x**2 - 4*x + 5

# Compute the indefinite integral
integral = sp.integrate(g, x)

# Print the integral
print("The indefinite integral of g(x) is:")
print(integral)

Explanation:

  1. Function Definition:

    • We define the function $ g(x) = 3x^2 - 4x + 5 $.

  2. Integration:

    • sp.integrate(g, x) computes the indefinite integral of g with respect to x.

  3. Output:

    • The indefinite integral of the function is printed, including the constant of integration.

Result:

1
2
The indefinite integral of g(x) is:
x**3 - 2*x**2 + 5*x

Conclusion

These examples illustrate how to solve equations, differentiate, and integrate using $SymPy$.

The library is powerful for symbolic math, allowing you to handle complex mathematical expressions and operations programmatically.