phase cycling.py

# tags for phase cycles
phase_cycles = [0, 1, 0.6, 1.6, 1.3, 2.3]

# subspaces
sub1, sub2 = np.zeros((3,3), dtype=complex), np.zeros((3,3), dtype=complex)
sub1 = np.array([[np.exp(1j*np.pi*(x)), np.exp(1j*np.pi*(-x)), 1] for x in phase_cycles[0::2]])
sub2 = np.array([[np.exp(1j*np.pi*(x)), np.exp(1j*np.pi*(-x)), 1] for x in phase_cycles[1::2]])

# Full matrix
C = np.kron(np.array([[1, 0], [0, 0]]), sub1) + np.kron(np.array([[0, 0], [0, 1]]), sub2)

# permutation matrix, because data taking is actually interleaved
perm = np.array([[1, 0, 0, 0, 0, 0],
                 [0, 0, 0, 1, 0, 0],
                 [0, 1, 0, 0, 0, 0],
                 [0, 0, 0, 0, 1, 0],
                 [0, 0, 1, 0, 0, 0],
                 [0, 0, 0, 0, 0, 1]])

Cperm = np.dot(perm, C) 

# final phase cycling matrix
Cperm_inv = np.linalg.inv(Cperm)