Adrien/QElephant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove complex i

Browse source
This commit is contained in:
Didictateur 2023-11-08 14:30:24 +01:00
parent 8dd2895752
commit a4c65223d9
1 changed files with 12 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
QElephant/Matrix.py
View file

@ -1,8 +1,6 @@
import math import math
import numpy as np import numpy as np
I = complex(0, 1)
class Matrix: class Matrix:
def __init__(self, l: list[list[complex]]=[]) -> None: def __init__(self, l: list[list[complex]]=[]) -> None:
if type(l) is not list: if type(l) is not list:
@ -72,15 +70,15 @@ class Matrix:
@staticmethod @staticmethod
def SQRTX() -> "Matrix": def SQRTX() -> "Matrix":
return Matrix([ return Matrix([
[(1+I)/2, (1-I)/2], [(1+complex(0, 1))/2, (1-complex(0, 1))/2],
[(1-I)/2, (1+I)/2] [(1-complex(0, 1))/2, (1+complex(0, 1))/2]
]) ])
@staticmethod @staticmethod
def Y() -> "Matrix": def Y() -> "Matrix":
return Matrix([ return Matrix([
[0, -I], [0, -complex(0, 1)],
[I, 0] [complex(0, 1), 0]
]) ])
@staticmethod @staticmethod
@ -94,12 +92,12 @@ class Matrix:
def S() -> "Matrix": def S() -> "Matrix":
return Matrix([ return Matrix([
[1, 0], [1, 0],
[0, I] [0, complex(0, 1)]
]) ])
@staticmethod @staticmethod
def T() -> "Matrix": def T() -> "Matrix":
s = math.e**(I*math.pi/4) s = math.e**(complex(0, 1)*math.pi/4)
return Matrix([ return Matrix([
[1, 0], [1, 0],
[0, s] [0, s]
@ -113,8 +111,8 @@ class Matrix:
c = math.cos(phi/2) c = math.cos(phi/2)
s = math.sin(phi/2) s = math.sin(phi/2)
return Matrix([ return Matrix([
[c, -I*s], [c, -complex(0, 1)*s],
[-I*s, c] [-complex(0, 1)*s, c]
]) ])
@staticmethod @staticmethod
@ -134,7 +132,7 @@ class Matrix:
if type(phi) not in {int, float}: if type(phi) not in {int, float}:
TypeError(f"an angle must be integer or float, not {type(phi)}") TypeError(f"an angle must be integer or float, not {type(phi)}")
s = math.e**(I*phi/2) s = math.e**(complex(0, 1)*phi/2)
return Matrix([ return Matrix([
[1/s, 0], [1/s, 0],
[0, s] [0, s]
@ -145,7 +143,7 @@ class Matrix:
if type(phi) not in {int, float}: if type(phi) not in {int, float}:
TypeError(f"an angle must be integer or float, not {type(phi)}") TypeError(f"an angle must be integer or float, not {type(phi)}")
s = math.e**(I*phi) s = math.e**(complex(0, 1)*phi)
return Matrix([ return Matrix([
[1, 0], [1, 0],
[0, s] [0, s]
@ -165,8 +163,8 @@ class Matrix:
return Matrix([ return Matrix([
[1, 0, 0, 0], [1, 0, 0, 0],
[0, 1, 0, 0], [0, 1, 0, 0],
[0, 0, 0, -I], [0, 0, 0, -complex(0, 1)],
[0, 0, I, 0] [0, 0, complex(0, 1), 0]
]) ])
@staticmethod @staticmethod