qm-states-in-sympy.py

import sympy
from sympy import I, sqrt, exp, pi
from sympy.physics.quantum import qapply
from sympy.physics.quantum import Ket, Bra
from sympy.physics.quantum import InnerProduct, OuterProduct
from sympy.physics.quantum.qubit import Qubit
from sympy.physics.quantum.dagger import Dagger

def inner_product(a,b):
    k0 = Qubit(0)  
    k1 = Qubit(1)  
    
    ketpsi = a*k0 + b*k1
    brapsi = Dagger(ketpsi)
    ip = brapsi * ketpsi
    return qapply(ip)

def ket_ip(a0,a1,b0,b1):
    k0 = Qubit(0)  
    k1 = Qubit(1)  
    
    ketpsi1 = a0*k0 + a1*k1
    ketpsi2 = b0*k0 + b1*k1

    brapsi1 = Dagger(ketpsi1)

    ip = brapsi1 * ketpsi2
    return qapply(ip)