Functions
Manipulation and Creation of States and Operators
Quantum States
- basis(dimensions, n=None, offset=None, *, dtype=None)[source]
Generates the vector representation of a Fock state.
- Parameters:
- dimensionsint or list of ints
Number of Fock states in Hilbert space. If a list, then the resultant object will be a tensor product over spaces with those dimensions.
- nint or list of ints, optional (default 0 for all dimensions)
Integer corresponding to desired number state, defaults to 0 for all dimensions if omitted. The shape must match
dimensions
, e.g. ifdimensions
is a list, thenn
must either be omitted or a list of equal length.- offsetint or list of ints, optional (default 0 for all dimensions)
The lowest number state that is included in the finite number state representation of the state in the relevant dimension.
- dtypetype or str
storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- state
qutip.Qobj
Qobj representing the requested number state
|n>
.
- state
Notes
A subtle incompatibility with the quantum optics toolbox: In QuTiP:
basis(N, 0) = ground state
but in the qotoolbox:
basis(N, 1) = ground state
Examples
>>> basis(5,2) Quantum object: dims = [[5], [1]], shape = (5, 1), type = ket Qobj data = [[ 0.+0.j] [ 0.+0.j] [ 1.+0.j] [ 0.+0.j] [ 0.+0.j]] >>> basis([2,2,2], [0,1,0]) Quantum object: dims = [[2, 2, 2], [1, 1, 1]], shape = (8, 1), type = ket Qobj data = [[0.] [0.] [1.] [0.] [0.] [0.] [0.] [0.]]
- bell_state(state='00', *, dtype=None)[source]
Returns the selected Bell state:
\[\begin{split}\begin{aligned} \lvert B_{00}\rangle &= \frac1{\sqrt2}(\lvert00\rangle+\lvert11\rangle)\\ \lvert B_{01}\rangle &= \frac1{\sqrt2}(\lvert00\rangle-\lvert11\rangle)\\ \lvert B_{10}\rangle &= \frac1{\sqrt2}(\lvert01\rangle+\lvert10\rangle)\\ \lvert B_{11}\rangle &= \frac1{\sqrt2}(\lvert01\rangle-\lvert10\rangle)\\ \end{aligned}\end{split}\]- Parameters:
- statestr [‘00’, ‘01’, 10, 11]
Which bell state to return
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- Bell_stateqobj
Bell state
- bra(seq, dim=2, *, dtype=None)[source]
Produces a multiparticle bra state for a list or string, where each element stands for state of the respective particle.
- Parameters:
- seqstr / list of ints or characters
Each element defines state of the respective particle. (e.g. [1,1,0,1] or a string “1101”). For qubits it is also possible to use the following conventions:
‘g’/’e’ (ground and excited state)
‘u’/’d’ (spin up and down)
‘H’/’V’ (horizontal and vertical polarization)
Note: for dimension > 9 you need to use a list.
- dimint (default: 2) / list of ints
Space dimension for each particle: int if there are the same, list if they are different.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- braqobj
Examples
>>> bra("10") Quantum object: dims = [[1, 1], [2, 2]], shape = [1, 4], type = bra Qobj data = [[ 0. 0. 1. 0.]]
>>> bra("Hue") Quantum object: dims = [[1, 1, 1], [2, 2, 2]], shape = [1, 8], type = bra Qobj data = [[ 0. 1. 0. 0. 0. 0. 0. 0.]]
>>> bra("12", 3) Quantum object: dims = [[1, 1], [3, 3]], shape = [1, 9], type = bra Qobj data = [[ 0. 0. 0. 0. 0. 1. 0. 0. 0.]]
>>> bra("31", [5, 2]) Quantum object: dims = [[1, 1], [5, 2]], shape = [1, 10], type = bra Qobj data = [[ 0. 0. 0. 0. 0. 0. 0. 1. 0. 0.]]
- coherent(N, alpha, offset=0, method=None, *, dtype=None)[source]
Generates a coherent state with eigenvalue alpha.
Constructed using displacement operator on vacuum state.
- Parameters:
- Nint
Number of Fock states in Hilbert space.
- alphafloat/complex
Eigenvalue of coherent state.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the state. Using a non-zero offset will make the default method ‘analytic’.
- methodstring {‘operator’, ‘analytic’}
Method for generating coherent state.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- stateqobj
Qobj quantum object for coherent state
Notes
Select method ‘operator’ (default) or ‘analytic’. With the ‘operator’ method, the coherent state is generated by displacing the vacuum state using the displacement operator defined in the truncated Hilbert space of size ‘N’. This method guarantees that the resulting state is normalized. With ‘analytic’ method the coherent state is generated using the analytical formula for the coherent state coefficients in the Fock basis. This method does not guarantee that the state is normalized if truncated to a small number of Fock states, but would in that case give more accurate coefficients.
Examples
>>> coherent(5,0.25j) Quantum object: dims = [[5], [1]], shape = [5, 1], type = ket Qobj data = [[ 9.69233235e-01+0.j ] [ 0.00000000e+00+0.24230831j] [ -4.28344935e-02+0.j ] [ 0.00000000e+00-0.00618204j] [ 7.80904967e-04+0.j ]]
- coherent_dm(N, alpha, offset=0, method='operator', *, dtype=None)[source]
Density matrix representation of a coherent state.
Constructed via outer product of
qutip.states.coherent
- Parameters:
- Nint
Number of Fock states in Hilbert space.
- alphafloat/complex
Eigenvalue for coherent state.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the state.
- methodstring {‘operator’, ‘analytic’}
Method for generating coherent density matrix.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- dmqobj
Density matrix representation of coherent state.
Notes
Select method ‘operator’ (default) or ‘analytic’. With the ‘operator’ method, the coherent density matrix is generated by displacing the vacuum state using the displacement operator defined in the truncated Hilbert space of size ‘N’. This method guarantees that the resulting density matrix is normalized. With ‘analytic’ method the coherent density matrix is generated using the analytical formula for the coherent state coefficients in the Fock basis. This method does not guarantee that the state is normalized if truncated to a small number of Fock states, but would in that case give more accurate coefficients.
Examples
>>> coherent_dm(3,0.25j) Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 0.93941695+0.j 0.00000000-0.23480733j -0.04216943+0.j ] [ 0.00000000+0.23480733j 0.05869011+0.j 0.00000000-0.01054025j] [-0.04216943+0.j 0.00000000+0.01054025j 0.00189294+0.j ]]
- enr_fock(dims, excitations, state, *, dtype=None)[source]
Generate the Fock state representation in a excitation-number restricted state space. The dims argument is a list of integers that define the number of quantums states of each component of a composite quantum system, and the excitations specifies the maximum number of excitations for the basis states that are to be included in the state space. The state argument is a tuple of integers that specifies the state (in the number basis representation) for which to generate the Fock state representation.
- Parameters:
- dimslist
A list of the dimensions of each subsystem of a composite quantum system.
- excitationsinteger
The maximum number of excitations that are to be included in the state space.
- statelist of integers
The state in the number basis representation.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- ketQobj
A Qobj instance that represent a Fock state in the exication-number- restricted state space defined by dims and exciations.
- enr_state_dictionaries(dims, excitations)[source]
Return the number of states, and lookup-dictionaries for translating a state tuple to a state index, and vice versa, for a system with a given number of components and maximum number of excitations.
- Parameters:
- dims: list
A list with the number of states in each sub-system.
- excitationsinteger
The maximum numbers of dimension
- Returns:
- nstates, state2idx, idx2state: integer, dict, list
The number of states nstates, a dictionary for looking up state indices from a state tuple, and a list containing the state tuples ordered by state indices. state2idx and idx2state are reverses of each other, i.e., state2idx[idx2state[idx]] = idx and idx2state[state2idx[state]] = state.
- enr_thermal_dm(dims, excitations, n, *, dtype=None)[source]
Generate the density operator for a thermal state in the excitation-number- restricted state space defined by the dims and exciations arguments. See the documentation for enr_fock for a more detailed description of these arguments. The temperature of each mode in dims is specified by the average number of excitatons n.
- Parameters:
- dimslist
A list of the dimensions of each subsystem of a composite quantum system.
- excitationsinteger
The maximum number of excitations that are to be included in the state space.
- ninteger
The average number of exciations in the thermal state. n can be a float (which then applies to each mode), or a list/array of the same length as dims, in which each element corresponds specifies the temperature of the corresponding mode.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- dmQobj
Thermal state density matrix.
- fock(dimensions, n=None, offset=None, *, dtype=None)[source]
Bosonic Fock (number) state.
Same as
qutip.states.basis
.- Parameters:
- dimensionsint or list of ints
Number of Fock states in Hilbert space. If a list, then the resultant object will be a tensor product over spaces with those dimensions.
- nint or list of ints, optional (default 0 for all dimensions)
Integer corresponding to desired number state, defaults to 0 for all dimensions if omitted. The shape must match
dimensions
, e.g. ifdimensions
is a list, thenn
must either be omitted or a list of equal length.- offsetint or list of ints, optional (default 0 for all dimensions)
The lowest number state that is included in the finite number state representation of the state in the relevant dimension.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- Requested number state \(\left|n\right>\).
Examples
>>> fock(4,3) Quantum object: dims = [[4], [1]], shape = [4, 1], type = ket Qobj data = [[ 0.+0.j] [ 0.+0.j] [ 0.+0.j] [ 1.+0.j]]
- fock_dm(dimensions, n=None, offset=None, *, dtype=None)[source]
Density matrix representation of a Fock state
Constructed via outer product of
qutip.states.fock
.- Parameters:
- dimensionsint or list of ints
Number of Fock states in Hilbert space. If a list, then the resultant object will be a tensor product over spaces with those dimensions.
- nint or list of ints, optional (default 0 for all dimensions)
Integer corresponding to desired number state, defaults to 0 for all dimensions if omitted. The shape must match
dimensions
, e.g. ifdimensions
is a list, thenn
must either be omitted or a list of equal length.- offsetint or list of ints, optional (default 0 for all dimensions)
The lowest number state that is included in the finite number state representation of the state in the relevant dimension.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- dmqobj
Density matrix representation of Fock state.
Examples
>>> fock_dm(3,1) Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 0.+0.j 0.+0.j 0.+0.j] [ 0.+0.j 1.+0.j 0.+0.j] [ 0.+0.j 0.+0.j 0.+0.j]]
- ghz_state(N=3, *, dtype=None)[source]
- Returns the N-qubit GHZ-state:
[ |00...00> + |11...11> ] / sqrt(2)
- Parameters:
- Nint (default=3)
Number of qubits in state
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- Gqobj
N-qubit GHZ-state
- ket(seq, dim=2, *, dtype=None)[source]
Produces a multiparticle ket state for a list or string, where each element stands for state of the respective particle.
- Parameters:
- seqstr / list of ints or characters
Each element defines state of the respective particle. (e.g. [1,1,0,1] or a string “1101”). For qubits it is also possible to use the following conventions: - ‘g’/’e’ (ground and excited state) - ‘u’/’d’ (spin up and down) - ‘H’/’V’ (horizontal and vertical polarization) Note: for dimension > 9 you need to use a list.
- dimint (default: 2) / list of ints
Space dimension for each particle: int if there are the same, list if they are different.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- ketqobj
Examples
>>> ket("10") Quantum object: dims = [[2, 2], [1, 1]], shape = [4, 1], type = ket Qobj data = [[ 0.] [ 0.] [ 1.] [ 0.]]
>>> ket("Hue") Quantum object: dims = [[2, 2, 2], [1, 1, 1]], shape = [8, 1], type = ket Qobj data = [[ 0.] [ 1.] [ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 0.]]
>>> ket("12", 3) Quantum object: dims = [[3, 3], [1, 1]], shape = [9, 1], type = ket Qobj data = [[ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 1.] [ 0.] [ 0.] [ 0.]]
>>> ket("31", [5, 2]) Quantum object: dims = [[5, 2], [1, 1]], shape = [10, 1], type = ket Qobj data = [[ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 1.] [ 0.] [ 0.]]
- ket2dm(Q)[source]
Takes input ket or bra vector and returns density matrix formed by outer product. This is completely identical to calling Q.proj().
- Parameters:
- Qqobj
Ket or bra type quantum object.
- Returns:
- dmqobj
Density matrix formed by outer product of Q.
Examples
>>> x=basis(3,2) >>> ket2dm(x) Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 0.+0.j 0.+0.j 0.+0.j] [ 0.+0.j 0.+0.j 0.+0.j] [ 0.+0.j 0.+0.j 1.+0.j]]
- maximally_mixed_dm(N, *, dtype=None)[source]
Returns the maximally mixed density matrix for a Hilbert space of dimension N.
- Parameters:
- Nint
Number of basis states in Hilbert space.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- dmqobj
Thermal state density matrix.
- phase_basis(N, m, phi0=0, *, dtype=None)[source]
Basis vector for the mth phase of the Pegg-Barnett phase operator.
- Parameters:
- Nint
Number of basis vectors in Hilbert space.
- mint
Integer corresponding to the mth discrete phase phi_m = phi0 + 2 * pi * m / N
- phi0float (default=0)
Reference phase angle.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- stateqobj
Ket vector for mth Pegg-Barnett phase operator basis state.
Notes
The Pegg-Barnett basis states form a complete set over the truncated Hilbert space.
- projection(N, n, m, offset=None, *, dtype=None)[source]
The projection operator that projects state \(\lvert m\rangle\) on state \(\lvert n\rangle\).
- Parameters:
- Nint
Number of basis states in Hilbert space.
- n, mfloat
The number states in the projection.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the projector.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Requested projection operator.
- qutrit_basis(*, dtype=None)[source]
Basis states for a three level system (qutrit)
- dtypetype or str
storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- qstatesarray
Array of qutrit basis vectors
- singlet_state(*, dtype=None)[source]
Returns the two particle singlet-state:
\[\lvert S\rangle = \frac1{\sqrt2}(\lvert01\rangle-\lvert10\rangle)\]that is identical to the fourth bell state.
- Parameters:
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- Bell_stateqobj
\(\lvert B_{11}\rangle\) Bell state
- spin_coherent(j, theta, phi, type='ket', *, dtype=None)[source]
Generate the coherent spin state \(\lvert \theta, \phi\rangle\).
- Parameters:
- jfloat
The spin of the state.
- thetafloat
Angle from z axis.
- phifloat
Angle from x axis.
- typestring {‘ket’, ‘bra’, ‘dm’}
Type of state to generate.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- stateqobj
Qobj quantum object for spin coherent state
- spin_state(j, m, type='ket', *, dtype=None)[source]
Generates the spin state \(\lvert j, m\rangle\), i.e. the eigenstate of the spin-j Sz operator with eigenvalue m.
- Parameters:
- jfloat
The spin of the state ().
- mint
Eigenvalue of the spin-j Sz operator.
- typestring {‘ket’, ‘bra’, ‘dm’}
Type of state to generate.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- stateqobj
Qobj quantum object for spin state
- state_index_number(dims, index)[source]
Return a quantum number representation given a state index, for a system of composite structure defined by dims.
Example
>>> state_index_number([2, 2, 2], 6) [1, 1, 0]
- Parameters:
- dimslist or array
The quantum state dimensions array, as it would appear in a Qobj.
- indexinteger
The index of the state in standard enumeration ordering.
- Returns:
- statetuple
The state number tuple corresponding to index index in standard enumeration ordering.
- state_number_enumerate(dims, excitations=None)[source]
An iterator that enumerates all the state number tuples (quantum numbers of the form (n1, n2, n3, …)) for a system with dimensions given by dims.
Example
>>> for state in state_number_enumerate([2,2]): >>> print(state) ( 0 0 ) ( 0 1 ) ( 1 0 ) ( 1 1 )
- Parameters:
- dimslist or array
The quantum state dimensions array, as it would appear in a Qobj.
- excitationsinteger (None)
Restrict state space to states with excitation numbers below or equal to this value.
- Returns:
- state_numbertuple
Successive state number tuples that can be used in loops and other iterations, using standard state enumeration by definition.
- state_number_index(dims, state)[source]
Return the index of a quantum state corresponding to state, given a system with dimensions given by dims.
Example
>>> state_number_index([2, 2, 2], [1, 1, 0]) 6
- Parameters:
- dimslist or array
The quantum state dimensions array, as it would appear in a Qobj.
- statelist
State number array.
- Returns:
- idxint
The index of the state given by state in standard enumeration ordering.
- state_number_qobj(dims, state, *, dtype=None)[source]
Return a Qobj representation of a quantum state specified by the state array state.
Example
>>> state_number_qobj([2, 2, 2], [1, 0, 1]) Quantum object: dims = [[2, 2, 2], [1, 1, 1]], shape = [8, 1], type = ket Qobj data = [[ 0.] [ 0.] [ 0.] [ 0.] [ 0.] [ 1.] [ 0.] [ 0.]]
- Parameters:
- dimslist or array
The quantum state dimensions array, as it would appear in a Qobj.
- statelist
State number array.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- state
qutip.Qobj
The state as a
qutip.Qobj
instance.
Note
Deprecated in QuTiP 5.0, use
basis
instead.- state
- thermal_dm(N, n, method='operator', *, dtype=None)[source]
Density matrix for a thermal state of n particles
- Parameters:
- Nint
Number of basis states in Hilbert space.
- nfloat
Expectation value for number of particles in thermal state.
- methodstring {‘operator’, ‘analytic’}
string
that sets the method used to generate the thermal state probabilities- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- dmqobj
Thermal state density matrix.
Notes
The ‘operator’ method (default) generates the thermal state using the truncated number operator
num(N)
. This is the method that should be used in computations. The ‘analytic’ method uses the analytic coefficients derived in an infinite Hilbert space. The analytic form is not necessarily normalized, if truncated too aggressively.Examples
>>> thermal_dm(5, 1) Quantum object: dims = [[5], [5]], shape = [5, 5], type = oper, isHerm = True Qobj data = [[ 0.51612903 0. 0. 0. 0. ] [ 0. 0.25806452 0. 0. 0. ] [ 0. 0. 0.12903226 0. 0. ] [ 0. 0. 0. 0.06451613 0. ] [ 0. 0. 0. 0. 0.03225806]]
>>> thermal_dm(5, 1, 'analytic') Quantum object: dims = [[5], [5]], shape = [5, 5], type = oper, isHerm = True Qobj data = [[ 0.5 0. 0. 0. 0. ] [ 0. 0.25 0. 0. 0. ] [ 0. 0. 0.125 0. 0. ] [ 0. 0. 0. 0.0625 0. ] [ 0. 0. 0. 0. 0.03125]]
- triplet_states(*, dtype=None)[source]
Returns a list of the two particle triplet-states:
\[\lvert T_1\rangle = \lvert11\rangle \lvert T_2\rangle = \frac1{\sqrt2}(\lvert01\rangle + \lvert10\rangle) \lvert T_3\rangle = \lvert00\rangle\]- Parameters:
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- trip_stateslist
2 particle triplet states
- w_state(N=3, *, dtype=None)[source]
- Returns the N-qubit W-state:
[ |100..0> + |010..0> + |001..0> + ... |000..1> ] / sqrt(n)
- Parameters:
- Nint (default=3)
Number of qubits in state
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- W
Qobj
N-qubit W-state
- W
- zero_ket(N, dims=None, *, dtype=None)[source]
Creates the zero ket vector with shape Nx1 and dimensions dims.
- Parameters:
- Nint
Hilbert space dimensionality
- dimslist
Optional dimensions if ket corresponds to a composite Hilbert space.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- zero_ketqobj
Zero ket on given Hilbert space.
Quantum Operators
This module contains functions for generating Qobj representation of a variety of commonly occuring quantum operators.
- charge(Nmax, Nmin=None, frac=1, *, dtype=None)[source]
Generate the diagonal charge operator over charge states from Nmin to Nmax.
- Parameters:
- Nmaxint
Maximum charge state to consider.
- Nminint (default = -Nmax)
Lowest charge state to consider.
- fracfloat (default = 1)
Specify fractional charge if needed.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- CQobj
Charge operator over [Nmin, Nmax].
Notes
New in version 3.2.
- commutator(A, B, kind='normal')[source]
Return the commutator of kind kind (normal, anti) of the two operators A and B.
- create(N, offset=0, *, dtype=None)[source]
Creation (raising) operator.
- Parameters:
- Nint
Dimension of Hilbert space.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Qobj for raising operator.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
Examples
>>> create(4) Quantum object: dims=[[4], [4]], shape=(4, 4), type='oper', isherm=False Qobj data = [[ 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j] [ 1.00000000+0.j 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j] [ 0.00000000+0.j 1.41421356+0.j 0.00000000+0.j 0.00000000+0.j] [ 0.00000000+0.j 0.00000000+0.j 1.73205081+0.j 0.00000000+0.j]]
- destroy(N, offset=0, *, dtype=None)[source]
Destruction (lowering) operator.
- Parameters:
- Nint
Dimension of Hilbert space.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Qobj for lowering operator.
Examples
>>> destroy(4) Quantum object: dims=[[4], [4]], shape=(4, 4), type='oper', isherm=False Qobj data = [[ 0.00000000+0.j 1.00000000+0.j 0.00000000+0.j 0.00000000+0.j] [ 0.00000000+0.j 0.00000000+0.j 1.41421356+0.j 0.00000000+0.j] [ 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j 1.73205081+0.j] [ 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j 0.00000000+0.j]]
- displace(N, alpha, offset=0, *, dtype=None)[source]
Single-mode displacement operator.
- Parameters:
- Nint
Dimension of Hilbert space.
- alphafloat/complex
Displacement amplitude.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Displacement operator.
Examples
>>> displace(4,0.25) Quantum object: dims = [[4], [4]], shape = [4, 4], type = oper, isHerm = False Qobj data = [[ 0.96923323+0.j -0.24230859+0.j 0.04282883+0.j -0.00626025+0.j] [ 0.24230859+0.j 0.90866411+0.j -0.33183303+0.j 0.07418172+0.j] [ 0.04282883+0.j 0.33183303+0.j 0.84809499+0.j -0.41083747+0.j] [ 0.00626025+0.j 0.07418172+0.j 0.41083747+0.j 0.90866411+0.j]]
- enr_destroy(dims, excitations, *, dtype=None)[source]
Generate annilation operators for modes in a excitation-number-restricted state space. For example, consider a system consisting of 4 modes, each with 5 states. The total hilbert space size is 5**4 = 625. If we are only interested in states that contain up to 2 excitations, we only need to include states such as
(0, 0, 0, 0) (0, 0, 0, 1) (0, 0, 0, 2) (0, 0, 1, 0) (0, 0, 1, 1) (0, 0, 2, 0) …
This function creates annihilation operators for the 4 modes that act within this state space:
a1, a2, a3, a4 = enr_destroy([5, 5, 5, 5], excitations=2)
From this point onwards, the annihiltion operators a1, …, a4 can be used to setup a Hamiltonian, collapse operators and expectation-value operators, etc., following the usual pattern.
- Parameters:
- dimslist
A list of the dimensions of each subsystem of a composite quantum system.
- excitationsinteger
The maximum number of excitations that are to be included in the state space.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- a_opslist of qobj
A list of annihilation operators for each mode in the composite quantum system described by dims.
- enr_identity(dims, excitations, *, dtype=None)[source]
Generate the identity operator for the excitation-number restricted state space defined by the dims and exciations arguments. See the docstring for enr_fock for a more detailed description of these arguments.
- Parameters:
- dimslist
A list of the dimensions of each subsystem of a composite quantum system.
- excitationsinteger
The maximum number of excitations that are to be included in the state space.
- statelist of integers
The state in the number basis representation.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
A Qobj instance that represent the identity operator in the exication-number-restricted state space defined by dims and exciations.
- fcreate(n_sites, site, dtype=None)[source]
Fermionic creation operator. We use the Jordan-Wigner transformation, making use of the Jordan-Wigner ZZ..Z strings, to construct this as follows:
\[a_j = \sigma_z^{\otimes j} \otimes (frac{sigma_x - i sigma_y}{2}) \otimes I^{\otimes N-j-1}\]- Parameters:
- n_sitesint
Number of sites in Fock space.
- siteint
The site in Fock space to add a fermion to. Corresponds to j in the above JW transform.
- Returns:
- operqobj
Qobj for raising operator.
Examples
>>> fcreate(2) Quantum object: dims = [[2, 2], [2, 2]], shape = (4, 4), type = oper, isherm = False Qobj data = [[0. 0. 0. 0.] [0. 0. 0. 0.] [1. 0. 0. 0.] [0. 1. 0. 0.]]
- fdestroy(n_sites, site, dtype=None)[source]
Fermionic destruction operator. We use the Jordan-Wigner transformation, making use of the Jordan-Wigner ZZ..Z strings, to construct this as follows:
\[a_j = \sigma_z^{\otimes j} \otimes (\frac{\sigma_x + i \sigma_y}{2}) \otimes I^{\otimes N-j-1}\]- Parameters:
- n_sitesint
Number of sites in Fock space.
- siteint (default 0)
The site in Fock space to add a fermion to. Corresponds to j in the above JW transform.
- Returns:
- operqobj
Qobj for destruction operator.
Examples
>>> fdestroy(2) Quantum object: dims=[[2 2], [2 2]], shape=(4, 4), type='oper', isherm=False Qobj data = [[0. 0. 1. 0.] [0. 0. 0. 1.] [0. 0. 0. 0.] [0. 0. 0. 0.]]
- identity(dimensions, *, dtype=None)
Identity operator.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Identity operator Qobj.
Examples
>>> qeye(3) Quantum object: dims = [[3], [3]], shape = (3, 3), type = oper, isherm = True Qobj data = [[ 1. 0. 0.] [ 0. 1. 0.] [ 0. 0. 1.]] >>> qeye([2,2]) Quantum object: dims = [[2, 2], [2, 2]], shape = (4, 4), type = oper, isherm = True Qobj data = [[1. 0. 0. 0.] [0. 1. 0. 0.] [0. 0. 1. 0.] [0. 0. 0. 1.]]
- jmat(j, which=None, *, dtype=None)[source]
Higher-order spin operators:
- Parameters:
- jfloat
Spin of operator
- whichstr
Which operator to return ‘x’,’y’,’z’,’+’,’-‘. If no args given, then output is [‘x’,’y’,’z’]
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- jmatQobj or tuple of Qobj
qobj
for requested spin operator(s).
Notes
If no ‘args’ input, then returns array of [‘x’,’y’,’z’] operators.
Examples
>>> jmat(1) [ Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 0. 0.70710678 0. ] [ 0.70710678 0. 0.70710678] [ 0. 0.70710678 0. ]] Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 0.+0.j 0.-0.70710678j 0.+0.j ] [ 0.+0.70710678j 0.+0.j 0.-0.70710678j] [ 0.+0.j 0.+0.70710678j 0.+0.j ]] Quantum object: dims = [[3], [3]], shape = [3, 3], type = oper, isHerm = True Qobj data = [[ 1. 0. 0.] [ 0. 0. 0.] [ 0. 0. -1.]]]
- momentum(N, offset=0, *, dtype=None)[source]
Momentum operator p=-1j/sqrt(2)*(a-a.dag())
- Parameters:
- Nint
Number of Fock states in Hilbert space.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Momentum operator as Qobj.
- num(N, offset=0, *, dtype=None)[source]
Quantum object for number operator.
- Parameters:
- Nint
The dimension of the Hilbert space.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- oper: qobj
Qobj for number operator.
Examples
>>> num(4) Quantum object: dims=[[4], [4]], shape=(4, 4), type='oper', isherm=True Qobj data = [[0 0 0 0] [0 1 0 0] [0 0 2 0] [0 0 0 3]]
- phase(N, phi0=0, *, dtype=None)[source]
Single-mode Pegg-Barnett phase operator.
- Parameters:
- Nint
Number of basis states in Hilbert space.
- phi0float
Reference phase.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Phase operator with respect to reference phase.
Notes
The Pegg-Barnett phase operator is Hermitian on a truncated Hilbert space.
- position(N, offset=0, *, dtype=None)[source]
Position operator x=1/sqrt(2)*(a+a.dag())
- Parameters:
- Nint
Number of Fock states in Hilbert space.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Position operator as Qobj.
- qdiags(diagonals, offsets=None, dims=None, shape=None, *, dtype=None)[source]
Constructs an operator from an array of diagonals.
- Parameters:
- diagonalssequence of array_like
Array of elements to place along the selected diagonals.
- offsetssequence of ints, optional
- Sequence for diagonals to be set:
k=0 main diagonal
k>0 kth upper diagonal
k<0 kth lower diagonal
- dimslist, optional
Dimensions for operator
- shapelist, tuple, optional
Shape of operator. If omitted, a square operator large enough to contain the diagonals is generated.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
Examples
>>> qdiags(sqrt(range(1, 4)), 1) Quantum object: dims = [[4], [4]], shape = [4, 4], type = oper, isherm = False Qobj data = [[ 0. 1. 0. 0. ] [ 0. 0. 1.41421356 0. ] [ 0. 0. 0. 1.73205081] [ 0. 0. 0. 0. ]]
- qeye(dimensions, *, dtype=None)[source]
Identity operator.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Identity operator Qobj.
Examples
>>> qeye(3) Quantum object: dims = [[3], [3]], shape = (3, 3), type = oper, isherm = True Qobj data = [[ 1. 0. 0.] [ 0. 1. 0.] [ 0. 0. 1.]] >>> qeye([2,2]) Quantum object: dims = [[2, 2], [2, 2]], shape = (4, 4), type = oper, isherm = True Qobj data = [[1. 0. 0. 0.] [0. 1. 0. 0.] [0. 0. 1. 0.] [0. 0. 0. 1.]]
- qutrit_ops(*, dtype=None)[source]
Operators for a three level system (qutrit).
- Parameters:
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opers: array
array of qutrit operators.
- qzero(dimensions, *, dtype=None)[source]
Zero operator.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- qzeroqobj
Zero operator Qobj.
- sigmam()[source]
Annihilation operator for Pauli spins.
Examples
>>> sigmam() Quantum object: dims = [[2], [2]], shape = [2, 2], type = oper, isHerm = False Qobj data = [[ 0. 0.] [ 1. 0.]]
- sigmap()[source]
Creation operator for Pauli spins.
Examples
>>> sigmap() Quantum object: dims = [[2], [2]], shape = [2, 2], type = oper, isHerm = False Qobj data = [[ 0. 1.] [ 0. 0.]]
- sigmax()[source]
Pauli spin 1/2 sigma-x operator
Examples
>>> sigmax() Quantum object: dims = [[2], [2]], shape = [2, 2], type = oper, isHerm = False Qobj data = [[ 0. 1.] [ 1. 0.]]
- sigmay()[source]
Pauli spin 1/2 sigma-y operator.
Examples
>>> sigmay() Quantum object: dims = [[2], [2]], shape = [2, 2], type = oper, isHerm = True Qobj data = [[ 0.+0.j 0.-1.j] [ 0.+1.j 0.+0.j]]
- sigmaz()[source]
Pauli spin 1/2 sigma-z operator.
Examples
>>> sigmaz() Quantum object: dims = [[2], [2]], shape = [2, 2], type = oper, isHerm = True Qobj data = [[ 1. 0.] [ 0. -1.]]
- spin_Jm(j, *, dtype=None)[source]
Spin-j annihilation operator
- Parameters:
- jfloat
Spin of operator
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
qobj
representation of the operator.
- spin_Jp(j, *, dtype=None)[source]
Spin-j creation operator
- Parameters:
- jfloat
Spin of operator
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
qobj
representation of the operator.
- spin_Jx(j, *, dtype=None)[source]
Spin-j x operator
- Parameters:
- jfloat
Spin of operator
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
qobj
representation of the operator.
- spin_Jy(j, *, dtype=None)[source]
Spin-j y operator
- Parameters:
- jfloat
Spin of operator
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
qobj
representation of the operator.
- spin_Jz(j, *, dtype=None)[source]
Spin-j z operator
- Parameters:
- jfloat
Spin of operator
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- opQobj
qobj
representation of the operator.
- squeeze(N, z, offset=0, *, dtype=None)[source]
Single-mode squeezing operator.
- Parameters:
- Nint
Dimension of hilbert space.
- zfloat/complex
Squeezing parameter.
- offsetint (default 0)
The lowest number state that is included in the finite number state representation of the operator.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- oper
qutip.Qobj
Squeezing operator.
- oper
Examples
>>> squeeze(4, 0.25) Quantum object: dims = [[4], [4]], shape = [4, 4], type = oper, isHerm = False Qobj data = [[ 0.98441565+0.j 0.00000000+0.j 0.17585742+0.j 0.00000000+0.j] [ 0.00000000+0.j 0.95349007+0.j 0.00000000+0.j 0.30142443+0.j] [-0.17585742+0.j 0.00000000+0.j 0.98441565+0.j 0.00000000+0.j] [ 0.00000000+0.j -0.30142443+0.j 0.00000000+0.j 0.95349007+0.j]]
- squeezing(a1, a2, z)[source]
Generalized squeezing operator.
\[S(z) = \exp\left(\frac{1}{2}\left(z^*a_1a_2 - za_1^\dagger a_2^\dagger\right)\right)\]- Parameters:
- a1
qutip.Qobj
Operator 1.
- a2
qutip.Qobj
Operator 2.
- zfloat/complex
Squeezing parameter.
- a1
- Returns:
- oper
qutip.Qobj
Squeezing operator.
- oper
- tunneling(N, m=1, *, dtype=None)[source]
Tunneling operator with elements of the form \(\\sum |N><N+m| + |N+m><N|\).
- Parameters:
- Nint
Number of basis states in Hilbert space.
- mint (default = 1)
Number of excitations in tunneling event.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- TQobj
Tunneling operator.
Notes
New in version 3.2.
Quantum Objects
The Quantum Object (Qobj) class, for representing quantum states and operators, and related functions.
- ptrace(Q, sel)[source]
Partial trace of the Qobj with selected components remaining.
- Parameters:
- Q
qutip.Qobj
Composite quantum object.
- selint/list
An
int
orlist
of components to keep after partial trace.
- Q
- Returns:
- oper
qutip.Qobj
Quantum object representing partial trace with selected components remaining.
- oper
Notes
This function is for legacy compatibility only. It is recommended to use the
ptrace()
Qobj method.
Random Operators and States
This module is a collection of random state and operator generators.
- rand_dm(dimensions, density=0.75, distribution='ginibre', *, eigenvalues=(), rank=None, seed=None, dtype=None)[source]
Creates a random density matrix of the desired dimensions.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce eitheroper
orsuper
depending on the passeddimensions
.- densityfloat
Density between [0,1] of output density matrix. Used by the “pure”, “eigen” and “herm”.
- distributionstr {“ginibre”, “hs”, “pure”, “eigen”, “uniform”}
Method used to obtain the density matrices.
“ginibre” : Ginibre random density operator of rank
rank
by using the algorithm of [BCSZ08].“hs” : Hilbert-Schmidt ensemble, equivalent to a full rank ginibre operator.
“pure” : Density matrix created from a random ket.
“eigen” : A density matrix with the given
eigenvalues
.“herm” : Build from a random hermitian matrix using
rand_herm
.
- eigenvaluesarray_like, optional
Eigenvalues of the output Hermitian matrix. The len must match the shape of the matrix.
- rankint, optional
When using the “ginibre” distribution, rank of the density matrix. Will default to a full rank operator when not provided.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Density matrix quantum operator.
- rand_herm(dimensions, density=0.3, distribution='fill', *, eigenvalues=(), seed=None, dtype=None)[source]
Creates a random sparse Hermitian quantum object.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- densityfloat, [0.30]
Density between [0,1] of output Hermitian operator.
- distributionstr {“fill”, “pos_def”, “eigen”}
Method used to obtain the density matrices.
“fill” : Uses \(H=0.5*(X+X^{+})\) where \(X\) is a randomly generated quantum operator with elements uniformly distributed between
[-1, 1] + [-1j, 1j]
.“eigen” : A density matrix with the given
eigenvalues
. It uses random complex Jacobi rotations to shuffle the operator.“pos_def” : Return a positive semi-definite matrix by diagonal dominance.
- eigenvaluesarray_like, optional
Eigenvalues of the output Hermitian matrix. The len must match the shape of the matrix.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- oper
qobj
Hermitian quantum operator.
- oper
Notes
If given a list of eigenvalues the object is created using complex Jacobi rotations. While this method is fast for small matrices, it should not be repeatedly used for generating matrices larger than ~1000x1000.
- rand_ket(dimensions, density=1, distribution='haar', *, seed=None, dtype=None)[source]
Creates a random ket vector.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- densityfloat, [1]
Density between [0,1] of output ket state when using the
fill
method.- distributionstr {“haar”, “fill”}
Method used to obtain the kets.
haar : Haar random pure state obtained by applying a Haar random unitary to a fixed pure state.
fill : Fill the ket with uniformly distributed random complex number.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Ket quantum state vector.
- rand_stochastic(dimensions, density=0.75, kind='left', *, seed=None, dtype=None)[source]
Generates a random stochastic matrix.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- densityfloat, [0.75]
Density between [0,1] of output density matrix.
- kindstr (Default = ‘left’)
Generate ‘left’ or ‘right’ stochastic matrix.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Quantum operator form of stochastic matrix.
- rand_super(dimensions, *, superrep='super', seed=None, dtype=None)[source]
Returns a randomly drawn superoperator acting on operators acting on N dimensions.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- superropstr, optional, {“super”}
Representation of the super operator
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- rand_super_bcsz(dimensions, enforce_tp=True, rank=None, *, superrep='super', seed=None, dtype=None)[source]
Returns a random superoperator drawn from the Bruzda et al ensemble for CPTP maps [BCSZ08]. Note that due to finite numerical precision, for ranks less than full-rank, zero eigenvalues may become slightly negative, such that the returned operator is not actually completely positive.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If an int is provided, it is understood as the Square root of the dimension of the superoperator to be returned, with the corresponding dims as
[[[N],[N]], [[N],[N]]]
. If provided as a list of ints, then the dimensions is understood as the space of density matrices this superoperator is applied to:dimensions=[2,2]
dims=[[[2,2],[2,2]], [[2,2],[2,2]]]
.- enforce_tpbool
If True, the trace-preserving condition of [BCSZ08] is enforced; otherwise only complete positivity is enforced.
- rankint or None
Rank of the sampled superoperator. If None, a full-rank superoperator is generated.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- superropstr, optional, {“super”}
representation of the
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- rhoQobj
A superoperator acting on vectorized dim × dim density operators, sampled from the BCSZ distribution.
- rand_unitary(dimensions, density=1, distribution='haar', *, seed=None, dtype=None)[source]
Creates a random sparse unitary quantum object.
- Parameters:
- dimensions(int) or (list of int) or (list of list of int)
Dimension of Hilbert space. If provided as a list of ints, then the dimension is the product over this list, but the
dims
property of the new Qobj are set to this list. This can produce either oper or super depending on the passed dimensions.- densityfloat, [1]
Density between [0,1] of output unitary operator.
- distribution[“haar”, “exp”]
Method used to obtain the unitary matrices.
haar : Haar random unitary matrix using the algorithm of [Mez07].
exp : Uses \(\exp(-iH)\), where H is a randomly generated Hermitian operator.
- seedint, SeedSequence, Generator, optional
Seed to create the random number generator or a pre prepared generator. When none is suplied, a default generator is used.
- dtypetype or str
Storage representation. Any data-layer known to qutip.data.to is accepted.
- Returns:
- operqobj
Unitary quantum operator.
Superoperators and Liouvillians
- lindblad_dissipator(a, b=None, data_only=False, chi=None)[source]
Lindblad dissipator (generalized) for a single pair of collapse operators (a, b), or for a single collapse operator (a) when b is not specified:
\[\mathcal{D}[a,b]\rho = a \rho b^\dagger - \frac{1}{2}a^\dagger b\rho - \frac{1}{2}\rho a^\dagger b\]- Parameters:
- aQobj or QobjEvo
Left part of collapse operator.
- bQobj or QobjEvo (optional)
Right part of collapse operator. If not specified, b defaults to a.
- chifloat [None]
In some systems it is possible to determine the statistical moments (mean, variance, etc) of the probability distribution of the occupation numbers of states by numerically evaluating the derivatives of the steady state occupation probability as a function of an artificial phase parameter
chi
which multiplies thea \rho a^dagger
term of the dissipator bye ^ (i * chi)
. The factore ^ (i * chi)
is introduced via the generating function of the statistical moments. For examples of the technique, see Full counting statistics of nano-electromechanical systems and Photon-mediated electron transport in hybrid circuit-QED. This parameter is deprecated and may be removed in QuTiP 5.- data_onlybool [False]
Return the data object instead of a Qobj
- Returns:
- Dqobj, QobjEvo
Lindblad dissipator superoperator.
- liouvillian(H=None, c_ops=None, data_only=False, chi=None)[source]
Assembles the Liouvillian superoperator from a Hamiltonian and a
list
of collapse operators.- Parameters:
- HQobj or QobjEvo (optional)
System Hamiltonian or Hamiltonian component of a Liouvillian. Considered 0 if not given.
- c_opsarray_like of Qobj or QobjEvo
A
list
orarray
of collapse operators.- data_onlybool [False]
Return the data object instead of a Qobj
- chiarray_like of float [None]
In some systems it is possible to determine the statistical moments (mean, variance, etc) of the probability distributions of occupation of various states by numerically evaluating the derivatives of the steady state occupation probability as a function of artificial phase parameters
chi
which are included in thelindblad_dissipator
for each collapse operator. See the documentation oflindblad_dissipator
for references and further details. This parameter is deprecated and may be removed in QuTiP 5.
- Returns:
- LQobj or QobjEvo
Liouvillian superoperator.
- operator_to_vector(op)[source]
Create a vector representation given a quantum operator in matrix form. The passed object should have a
Qobj.type
of ‘oper’ or ‘super’; this function is not designed for general-purpose matrix reshaping.- Parameters:
- opQobj or QobjEvo
Quantum operator in matrix form. This must have a type of ‘oper’ or ‘super’.
- Returns:
- Qobj or QobjEvo
The same object, but re-cast into a column-stacked-vector form of type ‘operator-ket’. The output is the same type as the passed object.
- spost(A)[source]
Superoperator formed from post-multiplication by operator A
- Parameters:
- AQobj or QobjEvo
Quantum operator for post multiplication.
- Returns:
- superQobj or QobjEvo
Superoperator formed from input qauntum object.
- spre(A)[source]
Superoperator formed from pre-multiplication by operator A.
- Parameters:
- AQobj or QobjEvo
Quantum operator for pre-multiplication.
- Returns:
- super :Qobj or QobjEvo
Superoperator formed from input quantum object.
- sprepost(A, B)[source]
Superoperator formed from pre-multiplication by A and post-multiplication by B.
- Parameters:
- AQobj or QobjEvo
Quantum operator for pre-multiplication.
- BQobj or QobjEvo
Quantum operator for post-multiplication.
- Returns:
- superQobj or QobjEvo
Superoperator formed from input quantum objects.
- vector_to_operator(op)[source]
Create a matrix representation given a quantum operator in vector form. The passed object should have a
Qobj.type
of ‘operator-ket’; this function is not designed for general-purpose matrix reshaping.- Parameters:
- opQobj or QobjEvo
Quantum operator in column-stacked-vector form. This must have a type of ‘operator-ket’.
- Returns:
- Qobj or QobjEvo
The same object, but re-cast into “standard” operator form. The output is the same type as the passed object.
Superoperator Representations
This module implements transformations between superoperator representations, including supermatrix, Kraus, Choi and Chi (process) matrix formalisms.
- kraus_to_choi(kraus_list)[source]
Take a list of Kraus operators and returns the Choi matrix for the channel represented by the Kraus operators in kraus_list
- to_chi(q_oper)[source]
Converts a Qobj representing a quantum map to a representation as a chi (process) matrix in the Pauli basis, such that the trace of the returned operator is equal to the dimension of the system.
- Parameters:
- q_operQobj
Superoperator to be converted to Chi representation. If
q_oper
istype="oper"
, then it is taken to act by conjugation, such thatto_chi(A) == to_chi(sprepost(A, A.dag()))
.
- Returns:
- chiQobj
A quantum object representing the same map as
q_oper
, such thatchi.superrep == "chi"
.
- Raises:
- TypeError: if the given quantum object is not a map, or cannot be converted
to Chi representation.
- to_choi(q_oper)[source]
Converts a Qobj representing a quantum map to the Choi representation, such that the trace of the returned operator is equal to the dimension of the system.
- Parameters:
- q_operQobj
Superoperator to be converted to Choi representation. If
q_oper
istype="oper"
, then it is taken to act by conjugation, such thatto_choi(A) == to_choi(sprepost(A, A.dag()))
.
- Returns:
- choiQobj
A quantum object representing the same map as
q_oper
, such thatchoi.superrep == "choi"
.
- Raises:
- TypeError: if the given quantum object is not a map, or cannot be converted
to Choi representation.
- to_kraus(q_oper, tol=1e-09)[source]
Converts a Qobj representing a quantum map to a list of quantum objects, each representing an operator in the Kraus decomposition of the given map.
- Parameters:
- q_operQobj
Superoperator to be converted to Kraus representation. If
q_oper
istype="oper"
, then it is taken to act by conjugation, such thatto_kraus(A) == to_kraus(sprepost(A, A.dag())) == [A]
.- tolFloat
Optional threshold parameter for eigenvalues/Kraus ops to be discarded. The default is to=1e-9.
- Returns:
- kraus_opslist of Qobj
A list of quantum objects, each representing a Kraus operator in the decomposition of
q_oper
.
- Raises:
- TypeError: if the given quantum object is not a map, or cannot be
decomposed into Kraus operators.
- to_stinespring(q_oper, threshold=1e-10)[source]
Converts a Qobj representing a quantum map $Lambda$ to a pair of partial isometries $A$ and $B$ such that $Lambda(X) = Tr_2(A X B^dagger)$ for all inputs $X$, where the partial trace is taken over a a new index on the output dimensions of $A$ and $B$.
For completely positive inputs, $A$ will always equal $B$ up to precision errors.
- Parameters:
- q_operQobj
Superoperator to be converted to a Stinespring pair.
- Returns:
- A, BQobj
Quantum objects representing each of the Stinespring matrices for the input Qobj.
- to_super(q_oper)[source]
Converts a Qobj representing a quantum map to the supermatrix (Liouville) representation.
- Parameters:
- q_operQobj
Superoperator to be converted to supermatrix representation. If
q_oper
istype="oper"
, then it is taken to act by conjugation, such thatto_super(A) == sprepost(A, A.dag())
.
- Returns:
- superopQobj
A quantum object representing the same map as
q_oper
, such thatsuperop.superrep == "super"
.
- Raises:
- TypeError
If the given quantum object is not a map, or cannot be converted to supermatrix representation.
Operators and Superoperator Dimensions
Internal use module for manipulating dims specifications.
- collapse_dims_oper(dims)[source]
Given the dimensions specifications for a ket-, bra- or oper-type Qobj, returns a dimensions specification describing the same shape by collapsing all composite systems. For instance, the bra-type dimensions specification
[[2, 3], [1]]
collapses to[[6], [1]]
.- Parameters:
- dimslist of lists of ints
Dimensions specifications to be collapsed.
- Returns:
- collapsed_dimslist of lists of ints
Collapsed dimensions specification describing the same shape such that
len(collapsed_dims[0]) == len(collapsed_dims[1]) == 1
.
- collapse_dims_super(dims)[source]
Given the dimensions specifications for an operator-ket-, operator-bra- or super-type Qobj, returns a dimensions specification describing the same shape by collapsing all composite systems. For instance, the super-type dimensions specification
[[[2, 3], [2, 3]], [[2, 3], [2, 3]]]
collapses to[[[6], [6]], [[6], [6]]]
.- Parameters:
- dimslist of lists of ints
Dimensions specifications to be collapsed.
- Returns:
- collapsed_dimslist of lists of ints
Collapsed dimensions specification describing the same shape such that
len(collapsed_dims[i][j]) == 1
fori
andj
inrange(2)
.
- deep_remove(l, *what)[source]
Removes scalars from all levels of a nested list.
Given a list containing a mix of scalars and lists, returns a list of the same structure, but where one or more scalars have been removed.
Examples
>>> deep_remove([[[[0, 1, 2]], [3, 4], [5], [6, 7]]], 0, 5) [[[[1, 2]], [3, 4], [], [6, 7]]]
- dims_idxs_to_tensor_idxs(dims, indices)[source]
Given the dims of a Qobj instance, and some indices into dims, returns the corresponding tensor indices. This helps resolve, for instance, that column-stacking for superoperators, oper-ket and oper-bra implies that the input and output tensor indices are reversed from their order in dims.
- Parameters:
- dimslist
Dimensions specification for a Qobj.
- indicesint, list or tuple
Indices to convert to tensor indices. Can be specified as a single index, or as a collection of indices. In the latter case, this can be nested arbitrarily deep. For instance, [0, [0, (2, 3)]].
- Returns:
- tens_indicesint, list or tuple
Container of the same structure as indices containing the tensor indices for each element of indices.
- dims_to_tensor_perm(dims)[source]
Given the dims of a Qobj instance, returns a list representing a permutation from the flattening of that dims specification to the corresponding tensor indices.
- Parameters:
- dimslist
Dimensions specification for a Qobj.
- Returns:
- permlist
A list such that
data[flatten(dims)[idx]]
gives the index of the tensordata
corresponding to theidx``th dimension of ``dims
.
- dims_to_tensor_shape(dims)[source]
Given the dims of a Qobj instance, returns the shape of the corresponding tensor. This helps, for instance, resolve the column-stacking convention for superoperators.
- Parameters:
- dimslist
Dimensions specification for a Qobj.
- Returns:
- tensor_shapetuple
NumPy shape of the corresponding tensor.
- enumerate_flat(l)[source]
Labels the indices at which scalars occur in a flattened list.
Given a list containing a mix of scalars and lists, returns a list of the same structure, where each scalar has been replaced by an index into the flattened list.
Examples
>>> print(enumerate_flat([[[10], [20, 30]], 40])) [[[0], [1, 2]], 3]
- flatten(l)[source]
Flattens a list of lists to the first level.
Given a list containing a mix of scalars and lists, flattens down to a list of the scalars within the original list.
Examples
>>> flatten([[[0], 1], 2]) [0, 1, 2]
- is_scalar(dims)[source]
Returns True if a dims specification is effectively a scalar (has dimension 1).
- unflatten(l, idxs)[source]
Unflattens a list by a given structure.
Given a list of scalars and a deep list of indices as produced by flatten, returns an “unflattened” form of the list. This perfectly inverts flatten.
Examples
>>> l = [[[10, 20, 30], [40, 50, 60]], [[70, 80, 90], [100, 110, 120]]] >>> idxs = enumerate_flat(l) >>> unflatten(flatten(l), idxs) == l True
Functions acting on states and operators
Expectation Values
- expect(oper, state)[source]
Calculate the expectation value for operator(s) and state(s). The expectation of state k on operator A is defined as k.dag() @ A @ k, and for density matrix R on operator A it is trace(A @ R).
- Parameters:
- operqobj/array-like
A single or a list or operators for expectation value.
- stateqobj/array-like
A single or a list of quantum states or density matrices.
- Returns:
- exptfloat/complex/array-like
Expectation value.
real
if oper is Hermitian,complex
otherwise. A (nested) array of expectaction values of state or operator are arrays.
Examples
>>> expect(num(4), basis(4, 3)) == 3 True
Tensor
Module for the creation of composite quantum objects via the tensor product.
- composite(*args)[source]
Given two or more operators, kets or bras, returns the Qobj corresponding to a composite system over each argument. For ordinary operators and vectors, this is the tensor product, while for superoperators and vectorized operators, this is the column-reshuffled tensor product.
If a mix of Qobjs supported on Hilbert and Liouville spaces are passed in, the former are promoted. Ordinary operators are assumed to be unitaries, and are promoted using
to_super
, while kets and bras are promoted by taking their projectors and usingoperator_to_vector(ket2dm(arg))
.
- super_tensor(*args)[source]
Calculate the tensor product of input superoperators, by tensoring together the underlying Hilbert spaces on which each vectorized operator acts.
- Parameters:
- argsarray_like
list
orarray
of quantum objects withtype="super"
.
- Returns:
- objqobj
A composite quantum object.
- tensor(*args)[source]
Calculates the tensor product of input operators.
- Parameters:
- argsarray_like
list
orarray
of quantum objects for tensor product.
- Returns:
- objqobj
A composite quantum object.
Examples
>>> tensor([sigmax(), sigmax()]) Quantum object: dims = [[2, 2], [2, 2]], shape = [4, 4], type = oper, isHerm = True Qobj data = [[ 0.+0.j 0.+0.j 0.+0.j 1.+0.j] [ 0.+0.j 0.+0.j 1.+0.j 0.+0.j] [ 0.+0.j 1.+0.j 0.+0.j 0.+0.j] [ 1.+0.j 0.+0.j 0.+0.j 0.+0.j]]
- tensor_contract(qobj, *pairs)[source]
Contracts a qobj along one or more index pairs. Note that this uses dense representations and thus should not be used for very large Qobjs.
- Parameters:
- pairstuple
One or more tuples
(i, j)
indicating that thei
andj
dimensions of the original qobj should be contracted.
- Returns:
- cqobjQobj
The original Qobj with all named index pairs contracted away.
Partial Transpose
- partial_transpose(rho, mask, method='dense')[source]
Return the partial transpose of a Qobj instance rho, where mask is an array/list with length that equals the number of components of rho (that is, the length of rho.dims[0]), and the values in mask indicates whether or not the corresponding subsystem is to be transposed. The elements in mask can be boolean or integers 0 or 1, where True/1 indicates that the corresponding subsystem should be tranposed.
- Parameters:
- rho
qutip.qobj
A density matrix.
- masklist / array
A mask that selects which subsystems should be transposed.
- methodstr
choice of method, dense or sparse. The default method is dense. The sparse implementation can be faster for large and sparse systems (hundreds of quantum states).
- rho
- Returns:
- rho_pr:
qutip.qobj
A density matrix with the selected subsystems transposed.
- rho_pr:
Entropy Functions
- concurrence(rho)[source]
Calculate the concurrence entanglement measure for a two-qubit state.
- Parameters:
- stateqobj
Ket, bra, or density matrix for a two-qubit state.
- Returns:
- concurfloat
Concurrence
References
- entropy_conditional(rho, selB, base=2.718281828459045, sparse=False)[source]
Calculates the conditional entropy \(S(A|B)=S(A,B)-S(B)\) of a selected density matrix component.
- Parameters:
- rhoqobj
Density matrix of composite object
- selBint/list
Selected components for density matrix B
- base{e,2}
Base of logarithm.
- sparse{False,True}
Use sparse eigensolver.
- Returns:
- ent_condfloat
Value of conditional entropy
- entropy_linear(rho)[source]
Linear entropy of a density matrix.
- Parameters:
- rhoqobj
sensity matrix or ket/bra vector.
- Returns:
- entropyfloat
Linear entropy of rho.
Examples
>>> rho=0.5*fock_dm(2,0)+0.5*fock_dm(2,1) >>> entropy_linear(rho) 0.5
- entropy_mutual(rho, selA, selB, base=2.718281828459045, sparse=False)[source]
Calculates the mutual information S(A:B) between selection components of a system density matrix.
- Parameters:
- rhoqobj
Density matrix for composite quantum systems
- selAint/list
int or list of first selected density matrix components.
- selBint/list
int or list of second selected density matrix components.
- base{e,2}
Base of logarithm.
- sparse{False,True}
Use sparse eigensolver.
- Returns:
- ent_mutfloat
Mutual information between selected components.
- entropy_relative(rho, sigma, base=2.718281828459045, sparse=False, tol=1e-12)[source]
Calculates the relative entropy S(rho||sigma) between two density matrices.
- Parameters:
- rho
qutip.Qobj
First density matrix (or ket which will be converted to a density matrix).
- sigma
qutip.Qobj
Second density matrix (or ket which will be converted to a density matrix).
- base{e,2}
Base of logarithm. Defaults to e.
- sparsebool
Flag to use sparse solver when determining the eigenvectors of the density matrices. Defaults to False.
- tolfloat
Tolerance to use to detect 0 eigenvalues or dot producted between eigenvectors. Defaults to 1e-12.
- rho
- Returns:
- rel_entfloat
Value of relative entropy. Guaranteed to be greater than zero and should equal zero only when rho and sigma are identical.
References
See Nielsen & Chuang, “Quantum Computation and Quantum Information”, Section 11.3.1, pg. 511 for a detailed explanation of quantum relative entropy.
Examples
First we define two density matrices:
>>> rho = qutip.ket2dm(qutip.ket("00")) >>> sigma = rho + qutip.ket2dm(qutip.ket("01")) >>> sigma = sigma.unit()
Then we calculate their relative entropy using base 2 (i.e.
log2
) and base e (i.e.log
).>>> qutip.entropy_relative(rho, sigma, base=2) 1.0 >>> qutip.entropy_relative(rho, sigma) 0.6931471805599453
- entropy_vn(rho, base=2.718281828459045, sparse=False)[source]
Von-Neumann entropy of density matrix
- Parameters:
- rhoqobj
Density matrix.
- base{e,2}
Base of logarithm.
- sparse{False,True}
Use sparse eigensolver.
- Returns:
- entropyfloat
Von-Neumann entropy of rho.
Examples
>>> rho=0.5*fock_dm(2,0)+0.5*fock_dm(2,1) >>> entropy_vn(rho,2) 1.0
Density Matrix Metrics
This module contains a collection of functions for calculating metrics (distance measures) between states and operators.
- average_gate_fidelity(oper, target=None)[source]
Returns the average gate fidelity of a quantum channel to the target channel, or to the identity channel if no target is given.
- Parameters:
- oper
qutip.Qobj
/list A unitary operator, or a superoperator in supermatrix, Choi or chi-matrix form, or a list of Kraus operators
- target
qutip.Qobj
A unitary operator
- oper
- Returns:
- fidfloat
Average gate fidelity between oper and target, or between oper and identity.
Notes
The average gate fidelity is defined for example in: A. Gilchrist, N.K. Langford, M.A. Nielsen, Phys. Rev. A 71, 062310 (2005). The definition of state fidelity that the average gate fidelity is based on is the one from R. Jozsa, Journal of Modern Optics, 41:12, 2315 (1994). It is the square of the fidelity implemented in
qutip.core.metrics.fidelity
which follows Nielsen & Chuang, “Quantum Computation and Quantum Information”
- bures_angle(A, B)[source]
Returns the Bures Angle between two density matrices A & B.
The Bures angle ranges from 0, for states with unit fidelity, to pi/2.
- Parameters:
- Aqobj
Density matrix or state vector.
- Bqobj
Density matrix or state vector with same dimensions as A.
- Returns:
- anglefloat
Bures angle between density matrices.
- bures_dist(A, B)[source]
Returns the Bures distance between two density matrices A & B.
The Bures distance ranges from 0, for states with unit fidelity, to sqrt(2).
- Parameters:
- Aqobj
Density matrix or state vector.
- Bqobj
Density matrix or state vector with same dimensions as A.
- Returns:
- distfloat
Bures distance between density matrices.
- dnorm(A, B=None, solver='CVXOPT', verbose=False, force_solve=False, sparse=True)[source]
Calculates the diamond norm of the quantum map q_oper, using the simplified semidefinite program of [Wat13].
The diamond norm SDP is solved by using CVXPY.
- Parameters:
- AQobj
Quantum map to take the diamond norm of.
- BQobj or None
If provided, the diamond norm of \(A - B\) is taken instead.
- solverstr
Solver to use with CVXPY. One of “CVXOPT” (default) or “SCS”. The latter tends to be significantly faster, but somewhat less accurate.
- verbosebool
If True, prints additional information about the solution.
- force_solvebool
If True, forces dnorm to solve the associated SDP, even if a special case is known for the argument.
- sparsebool
Whether to use sparse matrices in the convex optimisation problem. Default True.
- Returns:
- dnfloat
Diamond norm of q_oper.
- Raises:
- ImportError
If CVXPY cannot be imported.
- fidelity(A, B)[source]
Calculates the fidelity (pseudo-metric) between two density matrices. See: Nielsen & Chuang, “Quantum Computation and Quantum Information”
- Parameters:
- Aqobj
Density matrix or state vector.
- Bqobj
Density matrix or state vector with same dimensions as A.
- Returns:
- fidfloat
Fidelity pseudo-metric between A and B.
Examples
>>> x = fock_dm(5,3) >>> y = coherent_dm(5,1) >>> np.testing.assert_almost_equal(fidelity(x,y), 0.24104350624628332)
- hellinger_dist(A, B, sparse=False, tol=0)[source]
Calculates the quantum Hellinger distance between two density matrices.
Formula: hellinger_dist(A, B) = sqrt(2-2*Tr(sqrt(A)*sqrt(B)))
See: D. Spehner, F. Illuminati, M. Orszag, and W. Roga, “Geometric measures of quantum correlations with Bures and Hellinger distances” arXiv:1611.03449
- Parameters:
- A
qutip.Qobj
Density matrix or state vector.
- B
qutip.Qobj
Density matrix or state vector with same dimensions as A.
- tolfloat
Tolerance used by sparse eigensolver, if used. (0=Machine precision)
- sparse{False, True}
Use sparse eigensolver.
- A
- Returns:
- hellinger_distfloat
Quantum Hellinger distance between A and B. Ranges from 0 to sqrt(2).
Examples
>>> x=fock_dm(5,3) >>> y=coherent_dm(5,1) >>> np.testing.assert_almost_equal(hellinger_dist(x,y), 1.3725145002591095)
- hilbert_dist(A, B)[source]
Returns the Hilbert-Schmidt distance between two density matrices A & B.
- Parameters:
- Aqobj
Density matrix or state vector.
- Bqobj
Density matrix or state vector with same dimensions as A.
- Returns:
- distfloat
Hilbert-Schmidt distance between density matrices.
Notes
See V. Vedral and M. B. Plenio, Phys. Rev. A 57, 1619 (1998).
- process_fidelity(oper, target=None)[source]
Returns the process fidelity of a quantum channel to the target channel, or to the identity channel if no target is given. The process fidelity between two channels is defined as the state fidelity between their normalized Choi matrices.
- Parameters:
- oper
qutip.Qobj
/list A unitary operator, or a superoperator in supermatrix, Choi or chi-matrix form, or a list of Kraus operators
- target
qutip.Qobj
/list A unitary operator, or a superoperator in supermatrix, Choi or chi-matrix form, or a list of Kraus operators
- oper
- Returns:
- fidfloat
Process fidelity between oper and target, or between oper and identity.
Notes
Since Qutip 5.0, this function computes the process fidelity as defined for example in: A. Gilchrist, N.K. Langford, M.A. Nielsen, Phys. Rev. A 71, 062310 (2005). Previously, it computed a function that is now implemented in
control.fidcomp.FidCompUnitary.get_fidelity
. The definition of state fidelity that the process fidelity is based on is the one from R. Jozsa, Journal of Modern Optics, 41:12, 2315 (1994). It is the square of the one implemented inqutip.core.metrics.fidelity
which follows Nielsen & Chuang, “Quantum Computation and Quantum Information”
- tracedist(A, B, sparse=False, tol=0)[source]
Calculates the trace distance between two density matrices.. See: Nielsen & Chuang, “Quantum Computation and Quantum Information”
- Parameters:
- Aqobj
Density matrix or state vector.
- Bqobj
Density matrix or state vector with same dimensions as A.
- tolfloat
Tolerance used by sparse eigensolver, if used. (0=Machine precision)
- sparse{False, True}
Use sparse eigensolver.
- Returns:
- tracedistfloat
Trace distance between A and B.
Examples
>>> x=fock_dm(5,3) >>> y=coherent_dm(5,1) >>> np.testing.assert_almost_equal(tracedist(x,y), 0.9705143161472971)
Continuous Variables
This module contains a collection functions for calculating continuous variable quantities from fock-basis representation of the state of multi-mode fields.
- correlation_matrix(basis, rho=None)[source]
Given a basis set of operators \(\{a\}_n\), calculate the correlation matrix:
\[C_{mn} = \langle a_m a_n \rangle\]- Parameters:
- basislist
List of operators that defines the basis for the correlation matrix.
- rhoQobj
Density matrix for which to calculate the correlation matrix. If rho is None, then a matrix of correlation matrix operators is returned instead of expectation values of those operators.
- Returns:
- corr_matndarray
A 2-dimensional array of correlation values or operators.
- correlation_matrix_field(a1, a2, rho=None)[source]
Calculates the correlation matrix for given field operators \(a_1\) and \(a_2\). If a density matrix is given the expectation values are calculated, otherwise a matrix with operators is returned.
- Parameters:
- a1Qobj
Field operator for mode 1.
- a2Qobj
Field operator for mode 2.
- rhoQobj
Density matrix for which to calculate the covariance matrix.
- Returns:
- cov_matndarray
Array of complex numbers or Qobj’s A 2-dimensional array of covariance values, or, if rho=0, a matrix of operators.
- correlation_matrix_quadrature(a1, a2, rho=None, g=1.4142135623730951)[source]
Calculate the quadrature correlation matrix with given field operators \(a_1\) and \(a_2\). If a density matrix is given the expectation values are calculated, otherwise a matrix with operators is returned.
- Parameters:
- a1Qobj
Field operator for mode 1.
- a2Qobj
Field operator for mode 2.
- rhoQobj
Density matrix for which to calculate the covariance matrix.
- gfloat
Scaling factor for a = 0.5 * g * (x + iy), default g = sqrt(2). The value of g is related to the value of hbar in the commutation relation [x, y] = i * hbar via hbar=2/g ** 2 giving the default value hbar=1.
- Returns:
- corr_matndarray
Array of complex numbers or Qobj’s A 2-dimensional array of covariance values for the field quadratures, or, if rho=0, a matrix of operators.
- covariance_matrix(basis, rho, symmetrized=True)[source]
Given a basis set of operators \(\{a\}_n\), calculate the covariance matrix:
\[V_{mn} = \frac{1}{2}\langle a_m a_n + a_n a_m \rangle - \langle a_m \rangle \langle a_n\rangle\]or, if of the optional argument symmetrized=False,
\[V_{mn} = \langle a_m a_n\rangle - \langle a_m \rangle \langle a_n\rangle\]- Parameters:
- basislist
List of operators that defines the basis for the covariance matrix.
- rhoQobj
Density matrix for which to calculate the covariance matrix.
- symmetrizedbool {True, False}
Flag indicating whether the symmetrized (default) or non-symmetrized correlation matrix is to be calculated.
- Returns:
- corr_matndarray
A 2-dimensional array of covariance values.
- logarithmic_negativity(V, g=1.4142135623730951)[source]
Calculates the logarithmic negativity given a symmetrized covariance matrix, see
qutip.continuous_variables.covariance_matrix
. Note that the two-mode field state that is described by V must be Gaussian for this function to applicable.- Parameters:
- V2d array
The covariance matrix.
- gfloat
Scaling factor for a = 0.5 * g * (x + iy), default g = sqrt(2). The value of g is related to the value of hbar in the commutation relation [x, y] = i * hbar via hbar=2/g ** 2 giving the default value hbar=1.
- Returns:
- Nfloat
The logarithmic negativity for the two-mode Gaussian state that is described by the the Wigner covariance matrix V.
- wigner_covariance_matrix(a1=None, a2=None, R=None, rho=None, g=1.4142135623730951)[source]
Calculates the Wigner covariance matrix \(V_{ij} = \frac{1}{2}(R_{ij} + R_{ji})\), given the quadrature correlation matrix \(R_{ij} = \langle R_{i} R_{j}\rangle - \langle R_{i}\rangle \langle R_{j}\rangle\), where \(R = (q_1, p_1, q_2, p_2)^T\) is the vector with quadrature operators for the two modes.
Alternatively, if R = None, and if annihilation operators a1 and a2 for the two modes are supplied instead, the quadrature correlation matrix is constructed from the annihilation operators before then the covariance matrix is calculated.
- Parameters:
- a1Qobj
Field operator for mode 1.
- a2Qobj
Field operator for mode 2.
- Rndarray
The quadrature correlation matrix.
- rhoQobj
Density matrix for which to calculate the covariance matrix.
- gfloat
Scaling factor for a = 0.5 * g * (x + iy), default g = sqrt(2). The value of g is related to the value of hbar in the commutation relation [x, y] = i * hbar via hbar=2/g ** 2 giving the default value hbar=1.
- Returns:
- cov_matndarray
A 2-dimensional array of covariance values.
Measurement
Measurement of quantum states
Module for measuring quantum objects.
- measure(state, ops)[source]
A dispatch method that provides measurement results handling both observable style measurements and projector style measurements (POVMs and PVMs).
For return signatures, please check:
measure_observable
for observable measurements.measure_povm
for POVM measurements.
- measure_observable(state, op)[source]
Perform a measurement specified by an operator on the given state.
This function simulates the classic quantum measurement described in many introductory texts on quantum mechanics. The measurement collapses the state to one of the eigenstates of the given operator and the result of the measurement is the corresponding eigenvalue.
- Parameters:
- Returns:
- measured_valuefloat
The result of the measurement (one of the eigenvalues of op).
- state
Qobj
The new state (a ket if a ket was given, otherwise a density matrix).
Examples
Measure the z-component of the spin of the spin-up basis state:
>>> measure_observable(basis(2, 0), sigmaz()) (1.0, Quantum object: dims = [[2], [1]], shape = (2, 1), type = ket Qobj data = [[-1.] [ 0.]])
Since the spin-up basis is an eigenstate of sigmaz, this measurement always returns 1 as the measurement result (the eigenvalue of the spin-up basis) and the original state (up to a global phase).
Measure the x-component of the spin of the spin-down basis state:
>>> measure_observable(basis(2, 1), sigmax()) (-1.0, Quantum object: dims = [[2], [1]], shape = (2, 1), type = ket Qobj data = [[-0.70710678] [ 0.70710678]])
This measurement returns 1 fifty percent of the time and -1 the other fifty percent of the time. The new state returned is the corresponding eigenstate of sigmax.
One may also perform a measurement on a density matrix. Below we perform the same measurement as above, but on the density matrix representing the pure spin-down state:
>>> measure_observable(ket2dm(basis(2, 1)), sigmax()) (-1.0, Quantum object: dims = [[2], [2]], shape = (2, 2), type = oper Qobj data = [[ 0.5 -0.5] [-0.5 0.5]])
The measurement result is the same, but the new state is returned as a density matrix.
- measure_povm(state, ops)[source]
Perform a measurement specified by list of POVMs.
This function simulates a POVM measurement. The measurement collapses the state to one of the resultant states of the measurement and returns the index of the operator corresponding to the collapsed state as well as the collapsed state.
- Parameters:
- state
Qobj
The ket or density matrix specifying the state to measure.
- opslist of
Qobj
List of measurement operators \(M_i\) or kets. Either:
specifying a POVM s.t. \(E_i = M_i^\dagger M_i\)
projection operators if ops correspond to projectors (s.t. \(E_i = M_i^\dagger = M_i\))
kets (transformed to projectors)
- state
- Returns:
- indexfloat
The resultant index of the measurement.
- state
Qobj
The new state (a ket if a ket was given, otherwise a density matrix).
- measurement_statistics(state, ops)[source]
A dispatch method that provides measurement statistics handling both observable style measurements and projector style measurements(POVMs and PVMs).
For return signatures, please check:
measurement_statistics_observable
for observable measurements.measurement_statistics_povm
for POVM measurements.
- Parameters:
- state
Qobj
The ket or density matrix specifying the state to measure.
- ops
Qobj
or list ofQobj
measurement observable (:class:.Qobj); or
list of measurement operators \(M_i\) or kets (list of
Qobj
) Either:specifying a POVM s.t. \(E_i = M_i^\dagger * M_i\)
projection operators if ops correspond to projectors (s.t. \(E_i = M_i^\dagger = M_i\))
kets (transformed to projectors)
- state
- measurement_statistics_observable(state, op)[source]
Return the measurement eigenvalues, eigenstates (or projectors) and measurement probabilities for the given state and measurement operator.
- Parameters:
- Returns:
- eigenvalues: list of float
The list of eigenvalues of the measurement operator.
- eigenstates_or_projectors: list of
Qobj
If the state was a ket, return the eigenstates of the measurement operator. Otherwise return the projectors onto the eigenstates.
- probabilities: list of float
The probability of measuring the state as being in the corresponding eigenstate (and the measurement result being the corresponding eigenvalue).
- measurement_statistics_povm(state, ops)[source]
Returns measurement statistics (resultant states and probabilities) for a measurement specified by a set of positive operator valued measurements on a specified ket or density matrix.
- Parameters:
- state
Qobj
The ket or density matrix specifying the state to measure.
- opslist of
Qobj
List of measurement operators \(M_i\) or kets. Either:
specifying a POVM s.t. \(E_i = M_i^\dagger M_i\)
projection operators if ops correspond to projectors (s.t. \(E_i = M_i^\dagger = M_i\))
kets (transformed to projectors)
- state
- Returns:
- collapsed_stateslist of
Qobj
The collapsed states obtained after measuring the qubits and obtaining the qubit specified by the target in the state specified by the index.
- probabilitieslist of floats
The probability of measuring a state in a the state specified by the index.
- collapsed_stateslist of
Dynamics and Time-Evolution
Schrödinger Equation
This module provides solvers for the unitary Schrodinger equation.
- sesolve(H, psi0, tlist, e_ops=None, args=None, options=None)[source]
Schrodinger equation evolution of a state vector or unitary matrix for a given Hamiltonian.
Evolve the state vector (psi0) using a given Hamiltonian (H), by integrating the set of ordinary differential equations that define the system. Alternatively evolve a unitary matrix in solving the Schrodinger operator equation.
The output is either the state vector or unitary matrix at arbitrary points in time (tlist), or the expectation values of the supplied operators (e_ops). If e_ops is a callback function, it is invoked for each time in tlist with time and the state as arguments, and the function does not use any return values. e_ops cannot be used in conjunction with solving the Schrodinger operator equation
Time-dependent operators
For time-dependent problems, H and c_ops can be a
QobjEvo
or object that can be interpreted asQobjEvo
such as a list of (Qobj, Coefficient) pairs or a function.- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. System Hamiltonian as a Qobj or QobjEvo for time-dependent Hamiltonians. List of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- psi0
qutip.qobj
initial state vector (ket) or initial unitary operator psi0 = U
- tlistlist / array
list of times for \(t\).
- e_ops
qutip.qobj
, callable, or list. Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- argsNone / dictionary
dictionary of parameters for time-dependent Hamiltonians
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’} How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str [“adams”, “bdf”, “lsoda”, “dop853”, “vern9”, etc.] Which differential equation integration method to use.
atol, rtol : float Absolute and relative tolerance of the ODE integrator.
nsteps : Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, 0 Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
Other options could be supported depending on the integration method, see Integrator.
- H
- Returns:
- result:
qutip.Result
An instance of the class
qutip.Result
, which contains a list of array result.expect of expectation values for the times specified by tlist, and/or a list result.states of state vectors or density matrices corresponding to the times in tlist [if e_ops is an empty list of store_states=True in options].
- result:
Master Equation
This module provides solvers for the Lindblad master equation and von Neumann equation.
- mesolve(H, rho0, tlist, c_ops=None, e_ops=None, args=None, options=None)[source]
Master equation evolution of a density matrix for a given Hamiltonian and set of collapse operators, or a Liouvillian.
Evolve the state vector or density matrix (rho0) using a given Hamiltonian or Liouvillian (H) and an optional set of collapse operators (c_ops), by integrating the set of ordinary differential equations that define the system. In the absence of collapse operators the system is evolved according to the unitary evolution of the Hamiltonian.
The output is either the state vector at arbitrary points in time (tlist), or the expectation values of the supplied operators (e_ops). If e_ops is a callback function, it is invoked for each time in tlist with time and the state as arguments, and the function does not use any return values.
If either H or the Qobj elements in c_ops are superoperators, they will be treated as direct contributions to the total system Liouvillian. This allows the solution of master equations that are not in standard Lindblad form.
Time-dependent operators
For time-dependent problems, H and c_ops can be a
QobjEvo
or object that can be interpreted asQobjEvo
such as a list of (Qobj, Coefficient) pairs or a function.Additional options
Additional options to mesolve can be set via the options argument. Many ODE integration options can be set this way, and the store_states and store_final_state options can be used to store states even though expectation values are requested via the e_ops argument.
Note
When no collapse operator are given and the H is not a superoperator, it will defer to
sesolve
.- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. Possibly time-dependent system Liouvillian or Hamiltonian as a Qobj or QobjEvo. List of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- rho0
Qobj
initial density matrix or state vector (ket).
- tlistlist / array
list of times for \(t\).
- c_opslist of (
QobjEvo
,QobjEvo
compatible format) Single collapse operator, or list of collapse operators, or a list of Liouvillian superoperators. None is equivalent to an empty list.
- e_opslist of
Qobj
/ callback function Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- argsNone / dictionary
dictionary of parameters for time-dependent Hamiltonians and collapse operators.
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’} How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str [“adams”, “bdf”, “lsoda”, “dop853”, “vern9”, etc.] Which differential equation integration method to use.
atol, rtol : float Absolute and relative tolerance of the ODE integrator.
nsteps : Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, 0 Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
Other options could be supported depending on the integration method, see Integrator.
- H
- Returns:
- result:
qutip.Result
An instance of the class
qutip.Result
, which contains a list of array result.expect of expectation values for the times specified by tlist, and/or a list result.states of state vectors or density matrices corresponding to the times in tlist [if e_ops is an empty list of store_states=True in options].
- result:
Monte Carlo Evolution
- mcsolve(H, state, tlist, c_ops=(), e_ops=None, ntraj=500, *, args=None, options=None, seeds=None, target_tol=None, timeout=None)[source]
Monte Carlo evolution of a state vector \(|\psi \rangle\) for a given Hamiltonian and sets of collapse operators. Options for the underlying ODE solver are given by the Options class.
- Parameters:
- H
qutip.Qobj
,qutip.QobjEvo
,list
, callable. System Hamiltonian as a Qobj, QobjEvo. It can also be any input type that QobjEvo accepts (see
qutip.QobjEvo
’s documentation).H
can also be a superoperator (liouvillian) if some collapse operators are to be treated deterministically.- state
qutip.Qobj
Initial state vector.
- tlistarray_like
Times at which results are recorded.
- c_opslist
A
list
of collapse operators in any input type that QobjEvo accepts (seequtip.QobjEvo
’s documentation). They must be operators even ifH
is a superoperator. If none are given, the solver will defer tosesolve
ormesolve
.- e_opslist, [optional]
A
list
of operator as Qobj, QobjEvo or callable with signature of (t, state: Qobj) for calculating expectation values. When noe_ops
are given, the solver will default to save the states.- ntrajint
Maximum number of trajectories to run. Can be cut short if a time limit is passed with the
timeout
keyword or if the target tolerance is reached, seetarget_tol
.- argsNone / dict
Arguments for time-dependent Hamiltonian and collapse operator terms.
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool, [False] Whether or not to store the final state of the evolution in the result class.
store_states : bool, NoneType, [None] Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’}, [‘text’] How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict, [{“chunk_size”: 10}] kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str {“adams”, “bdf”, “dop853”, “vern9”, etc.}, [“adams”] Which differential equation integration method to use.
keep_runs_results : bool, [False] Whether to store results from all trajectories or just store the averages.
map : str {“serial”, “parallel”, “loky”}, [“serial”] How to run the trajectories. “parallel” uses concurent module to run in parallel while “loky” use the module of the same name to do so.
job_timeout : NoneType, int, [None] Maximum time to compute one trajectory.
num_cpus : NoneType, int, [None] Number of cpus to use when running in parallel.
None
detect the number of available cpus.norm_t_tol, norm_tol, norm_steps : float, float, int, [1e-6, 1e-4, 5] Parameters used to find the collapse location.
norm_t_tol
andnorm_tol
are the tolerance in time and norm respectively. An error will be raised if the collapse could not be found withinnorm_steps
tries.mc_corr_eps : float, [1e-10] Small number used to detect non-physical collapse caused by numerical imprecision.
atol, rtol : float, [1e-8, 1e-6] Absolute and relative tolerance of the ODE integrator.
nsteps : int [2500] Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, [0] Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
improved_sampling : Bool Whether to use the improved sampling algorithm from Abdelhafez et al. PRA (2019)
- seedsint, SeedSequence, list, [optional]
Seed for the random number generator. It can be a single seed used to spawn seeds for each trajectory or a list of seeds, one for each trajectory. Seeds are saved in the result and they can be reused with:
seeds=prev_result.seeds
- target_tolfloat, tuple, list, [optional]
Target tolerance of the evolution. The evolution will compute trajectories until the error on the expectation values is lower than this tolerance. The maximum number of trajectories employed is given by
ntraj
. The error is computed using jackknife resampling.target_tol
can be an absolute tolerance or a pair of absolute and relative tolerance, in that order. Lastly, it can be a list of pairs of (atol, rtol) for each e_ops.- timeoutfloat, [optional]
Maximum time for the evolution in second. When reached, no more trajectories will be computed.
- H
- Returns:
- results
qutip.solver.McResult
Object storing all results from the simulation. Which results is saved depends on the presence of
e_ops
and the options used.collapse
andphotocurrent
is available to Monte Carlo simulation results.
- results
- nm_mcsolve(H, state, tlist, ops_and_rates=(), e_ops=None, ntraj=500, *, args=None, options=None, seeds=None, target_tol=None, timeout=None)[source]
Monte-Carlo evolution corresponding to a Lindblad equation with “rates” that may be negative. Usage of this function is analogous to
mcsolve
, but thec_ops
parameter is replaced by anops_and_rates
parameter to allow for negative rates. Options for the underlying ODE solver are given by the Options class.- Parameters:
- H
qutip.Qobj
,qutip.QobjEvo
,list
, callable. System Hamiltonian as a Qobj, QobjEvo. It can also be any input type that QobjEvo accepts (see
qutip.QobjEvo
’s documentation).H
can also be a superoperator (liouvillian) if some collapse operators are to be treated deterministically.- state
qutip.Qobj
Initial state vector.
- tlistarray_like
Times at which results are recorded.
- ops_and_rateslist
A
list
of tuples(L, Gamma)
, where the Lindblad operatorL
is aqutip.Qobj
andGamma
represents the corresponding rate, which is allowed to be negative. The Lindblad operators must be operators even ifH
is a superoperator. If none are given, the solver will defer tosesolve
ormesolve
. Each rateGamma
may be just a number (in the case of a constant rate) or, otherwise, specified using any format accepted byqutip.coefficient
.- e_opslist, [optional]
A
list
of operator as Qobj, QobjEvo or callable with signature of (t, state: Qobj) for calculating expectation values. When noe_ops
are given, the solver will default to save the states.- ntrajint
Maximum number of trajectories to run. Can be cut short if a time limit is passed with the
timeout
keyword or if the target tolerance is reached, seetarget_tol
.- argsNone / dict
Arguments for time-dependent Hamiltonian and collapse operator terms.
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool, [False] Whether or not to store the final state of the evolution in the result class.
store_states : bool, NoneType, [None] Whether or not to store the state density matrices. On
None
the states will be saved if no expectation operators are given.progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’}, [‘text’] How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict, [{“chunk_size”: 10}] kwargs to pass to the progress_bar. Qutip’s bars use
chunk_size
.method : str {“adams”, “bdf”, “dop853”, “vern9”, etc.}, [“adams”] Which differential equation integration method to use.
keep_runs_results : bool, [False] Whether to store results from all trajectories or just store the averages.
map : str {“serial”, “parallel”, “loky”}, [“serial”] How to run the trajectories. “parallel” uses concurrent module to run in parallel while “loky” use the module of the same name to do so.
job_timeout : NoneType, int, [None] Maximum time to compute one trajectory.
num_cpus : NoneType, int, [None] Number of cpus to use when running in parallel.
None
detect the number of available cpus.norm_t_tol, norm_tol, norm_steps : float, float, int, [1e-6, 1e-4, 5] Parameters used to find the collapse location.
norm_t_tol
andnorm_tol
are the tolerance in time and norm respectively. An error will be raised if the collapse could not be found withinnorm_steps
tries.mc_corr_eps : float, [1e-10] Small number used to detect non-physical collapse caused by numerical imprecision.
atol, rtol : float, [1e-8, 1e-6] Absolute and relative tolerance of the ODE integrator.
nsteps : int [2500] Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, [0] Maximum length of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
completeness_rtol, completeness_atol : float, float, [1e-5, 1e-8] Parameters used in determining whether the given Lindblad operators satisfy a certain completeness relation. If they do not, an additional Lindblad operator is added automatically (with zero rate).
martingale_quad_limit : float or int, [100] An upper bound on the number of subintervals used in the adaptive integration of the martingale.
- seedsint, SeedSequence, list, [optional]
Seed for the random number generator. It can be a single seed used to spawn seeds for each trajectory or a list of seeds, one for each trajectory. Seeds are saved in the result and they can be reused with:
seeds=prev_result.seeds
- target_tolfloat, tuple, list, [optional]
Target tolerance of the evolution. The evolution will compute trajectories until the error on the expectation values is lower than this tolerance. The maximum number of trajectories employed is given by
ntraj
. The error is computed using jackknife resampling.target_tol
can be an absolute tolerance or a pair of absolute and relative tolerance, in that order. Lastly, it can be a list of pairs of (atol, rtol) for each e_ops.- timeoutfloat, [optional]
Maximum time for the evolution in seconds. When reached, no more trajectories will be computed.
- H
- Returns:
- results
qutip.solver.NmmcResult
Object storing all results from the simulation. Compared to a result returned by
mcsolve
, this result contains the additional fieldtrace
(andruns_trace
ifstore_final_state
is set). Note that the states on the individual trajectories are not normalized. This field contains the average of their trace, which will converge to one in the limit of sufficiently many trajectories.
- results
Krylov Subspace Solver
- krylovsolve(H, psi0, tlist, krylov_dim, e_ops=None, args=None, options=None)[source]
Schrodinger equation evolution of a state vector for time independent Hamiltonians using Krylov method.
Evolve the state vector (“psi0”) finding an approximation for the time evolution operator of Hamiltonian (“H”) by obtaining the projection of the time evolution operator on a set of small dimensional Krylov subspaces (m << dim(H)).
The output is either the state vector or unitary matrix at arbitrary points in time (tlist), or the expectation values of the supplied operators (e_ops). If e_ops is a callback function, it is invoked for each time in tlist with time and the state as arguments, and the function does not use any return values. e_ops cannot be used in conjunction with solving the Schrodinger operator equation
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. System Hamiltonian as a Qobj or QobjEvo for time-dependent Hamiltonians. List of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- psi0
qutip.qobj
initial state vector (ket) or initial unitary operator psi0 = U
- tlistlist / array
list of times for \(t\).
- krylov_dim: int
Dimension of Krylov approximation subspaces used for the time evolution approximation.
- e_ops
qutip.qobj
, callable, or list. Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- argsNone / dictionary
dictionary of parameters for time-dependent Hamiltonians
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool, [False] Whether or not to store the final state of the evolution in the result class.
store_states : bool, [None] Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool, [True] Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’}, [“text”] How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict, [{“chunk_size”: 10}] kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
atol: float [1e-7] Absolute and relative tolerance of the ODE integrator.
nsteps : int [100] Maximum number of (internally defined) steps allowed in one
tlist
step.min_step, max_step : float, [1e-5, 1e5] Miniumum and maximum lenght of one internal step.
always_compute_step: bool [False] If True, the step lenght is computed each time a new Krylov subspace is computed. Otherwise it is computed only once when creating the integrator.
sub_system_tol: float, [1e-7] Tolerance to detect an happy breakdown. An happy breakdown happens when the initial ket is in a subspace of the Hamiltonian smaller than
krylov_dim
.
- H
- Returns:
- result:
qutip.Result
An instance of the class
qutip.Result
, which contains a list of array result.expect of expectation values for the times specified by tlist, and/or a list result.states of state vectors or density matrices corresponding to the times in tlist [if e_ops is an empty list of store_states=True in options].
- result:
Bloch-Redfield Master Equation
This module provides solvers for the Lindblad master equation and von Neumann equation.
- brmesolve(H, psi0, tlist, a_ops=[], e_ops=[], c_ops=[], args={}, sec_cutoff=0.1, options=None)[source]
Solves for the dynamics of a system using the Bloch-Redfield master equation, given an input Hamiltonian, Hermitian bath-coupling terms and their associated spectral functions, as well as possible Lindblad collapse operators.
- Parameters:
- H
Qobj
,QobjEvo
Possibly time-dependent system Liouvillian or Hamiltonian as a Qobj or QobjEvo. list of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- psi0: Qobj
Initial density matrix or state vector (ket).
- tlistarray_like
List of times for evaluating evolution
- a_opslist of (a_op, spectra)
Nested list of system operators that couple to the environment, and the corresponding bath spectra.
- a_op
qutip.Qobj
,qutip.QobjEvo
The operator coupling to the environment. Must be hermitian.
- spectra
Coefficient
, str, func The corresponding bath spectral responce. Can be a Coefficient using an ‘w’ args, a function of the frequence or a string. Coefficient build from a numpy array are understood as a function of
w
instead oft
. Function are expected to be of the signaturef(w)
orf(t, w, **args)
.The spectra function can depend on
t
if the correspondinga_op
is aQobjEvo
.
Example:
a_ops = [ (a+a.dag(), ('w>0', args={"w": 0})), (QobjEvo(a+a.dag()), 'w > exp(-t)'), (QobjEvo([b+b.dag(), lambda t: ...]), lambda w: ...)), (c+c.dag(), SpectraCoefficient(coefficient(array, tlist=ws))), ]
- a_op
- e_opslist of
Qobj
/ callback function Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation- c_opslist of (
QobjEvo
,QobjEvo
compatible format) List of collapse operators.
- argsdict
Dictionary of parameters for time-dependent Hamiltonians and collapse operators. The key
w
is reserved for the spectra function.- sec_cutofffloat {0.1}
Cutoff for secular approximation. Use
-1
if secular approximation is not used when evaluating bath-coupling terms.- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’} How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
tensor_type : str [‘sparse’, ‘dense’, ‘data’] Which data type to use when computing the brtensor. With a cutoff ‘sparse’ is usually the most efficient.
sparse_eigensolver : bool {False} Whether to use the sparse eigensolver
method : str [“adams”, “bdf”, “lsoda”, “dop853”, “vern9”, etc.] Which differential equation integration method to use.
atol, rtol : float Absolute and relative tolerance of the ODE integrator.
nsteps : Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, 0 Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
Other options could be supported depending on the integration method, see Integrator.
- H
- Returns:
- result:
qutip.solver.Result
An instance of the class
qutip.solver.Result
, which contains either an array of expectation values, for operators given in e_ops, or a list of states for the times specified by tlist.
- result:
Floquet States and Floquet-Markov Master Equation
- class FMESolver(floquet_basis, a_ops, w_th=0.0, *, kmax=5, nT=None, options=None)[source]
Solver for the Floquet-Markov master equation.
Note
Operators (
c_ops
ande_ops
) are in the laboratory basis.- Parameters:
- floquet_basis
qutip.FloquetBasis
The system Hamiltonian wrapped in a FloquetBasis object. Choosing a different integrator for the
floquet_basis
than for the evolution of the floquet state can improve the performance.- a_opslist of tuple(
qutip.Qobj
, callable) List of collapse operators and the corresponding function for the noise power spectrum. The collapse operator must be a
Qobj
and cannot be time dependent. The spectrum function must take and return an numpy array.- w_thfloat
The temperature of the environment in units of Hamiltonian frequency.
- kmaxint [5]
The truncation of the number of sidebands..
- nTint [20*kmax]
The number of integration steps (for calculating X) within one period.
- optionsdict, optional
Options for the solver, see
FMESolver.options
and Integrator for a list of all options.
- floquet_basis
- resultclass
alias of
FloquetResult
- run(state0, tlist, *, floquet=False, args=None, e_ops=None)[source]
Calculate the evolution of the quantum system.
For a
state0
at timetlist[0]
do the evolution as directed byrhs
and for each time intlist
store the state and/or expectation values in aResult
. The evolution method and stored results are determined byoptions
.- Parameters:
- state0
Qobj
Initial state of the evolution.
- tlistlist of double
Time for which to save the results (state and/or expect) of the evolution. The first element of the list is the initial time of the evolution. Each times of the list must be increasing, but does not need to be uniformy distributed.
- floquetbool, optional {False}
Whether the initial state in the floquet basis or laboratory basis.
- argsdict, optional {None}
Not supported
- e_opslist {None}
List of Qobj, QobjEvo or callable to compute the expectation values. Function[s] must have the signature f(t : float, state : Qobj) -> expect.
- state0
- Returns:
- results
qutip.solver.FloquetResult
Results of the evolution. States and/or expect will be saved. You can control the saved data in the options.
- results
- start(state0, t0, *, floquet=False)[source]
Set the initial state and time for a step evolution.
options
for the evolutions are read at this step.- Parameters:
- state0
Qobj
Initial state of the evolution.
- t0double
Initial time of the evolution.
- floquetbool, optional {False}
Whether the initial state is in the floquet basis or laboratory basis.
- state0
- step(t, *, args=None, copy=True, floquet=False)[source]
Evolve the state to
t
and return the state as aQobj
.- Parameters:
- tdouble
Time to evolve to, must be higher than the last call.
- copybool, optional {True}
Whether to return a copy of the data or the data in the ODE solver.
- floquetbool, optional {False}
Whether to return the state in the floquet basis or laboratory basis.
- argsdict, optional {None}
Not supported
- .. note::
The state must be initialized first by calling
start
orrun
. Ifrun
is called,step
will continue from the last time and state obtained.
- class FloquetBasis(H, T, args=None, options=None, sparse=False, sort=True, precompute=None)[source]
Utility to compute floquet modes and states.
- Attributes:
- U
Propagator
The propagator of the Hamiltonian over one period.
- evecs
qutip.data.Data
Matrix where each column is an initial Floquet mode.
- e_quasinp.ndarray[float]
The quasi energies of the Hamiltonian.
- U
- from_floquet_basis(floquet_basis, t=0)[source]
Transform a ket or density matrix from the Floquet basis at time
t
to the lab basis.- Parameters:
- floquet_basis
Qobj
,qutip.data.Data
Initial state in the Floquet basis at time
t
. May be either a ket or density matrix.- tfloat [0]
The time at which to evaluate the Floquet states.
- floquet_basis
- Returns:
- output
Qobj
,qutip.data.Data
The state in the lab basis. The return type is the same as the type of the input state.
- output
- mode(t, data=False)[source]
Calculate the Floquet modes at time
t
.- Parameters:
- tfloat
The time for which to evaluate the Floquet mode.
- databool [False]
Whether to return the states as a single data matrix or a list of ket states.
- Returns:
- outputlist[
Qobj
],qutip.data.Data
A list of Floquet states for the time
t
or the states as column in a single matrix.
- outputlist[
- state(t, data=False)[source]
Evaluate the floquet states at time t.
- Parameters:
- tfloat
The time for which to evaluate the Floquet states.
- databool [False]
Whether to return the states as a single data matrix or a list of ket states.
- Returns:
- outputlist[
Qobj
],qutip.data.Data
A list of Floquet states for the time
t
or the states as column in a single matrix.
- outputlist[
- to_floquet_basis(lab_basis, t=0)[source]
Transform a ket or density matrix in the lab basis to the Floquet basis at time
t
.- Parameters:
- lab_basis
Qobj
,qutip.data.Data
Initial state in the lab basis.
- tfloat [0]
The time at which to evaluate the Floquet states.
- lab_basis
- Returns:
- output
Qobj
,qutip.data.Data
The state in the Floquet basis. The return type is the same as the type of the input state.
- output
- floquet_tensor(H, c_ops, spectra_cb, T=0, w_th=0.0, kmax=5, nT=100)[source]
Construct a tensor that represents the master equation in the floquet basis.
Simplest RWA approximation [Grifoni et al, Phys.Rep. 304 229 (1998)]
- Parameters:
- H
QobjEvo
Periodic Hamiltonian
- Tfloat
The period of the time-dependence of the hamiltonian.
- c_opslist of
qutip.qobj
list of collapse operators.
- spectra_cblist callback functions
List of callback functions that compute the noise power spectrum as a function of frequency for the collapse operators in c_ops.
- w_thfloat
The temperature in units of frequency.
- kmaxint
The truncation of the number of sidebands (default 5).
- H
- Returns:
- outputarray
The Floquet-Markov master equation tensor R.
- fmmesolve(H, rho0, tlist, c_ops=None, e_ops=None, spectra_cb=None, T=0, w_th=0.0, args=None, options=None)[source]
Solve the dynamics for the system using the Floquet-Markov master equation.
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. Periodic system Hamiltonian as
QobjEvo
. List of [Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- rho0 / psi0
qutip.Qobj
Initial density matrix or state vector (ket).
- tlistlist / array
List of times for \(t\).
- c_opslist of
qutip.Qobj
List of collapse operators. Time dependent collapse operators are not supported.
- e_opslist of
qutip.Qobj
/ callback function List of operators for which to evaluate expectation values. The states are reverted to the lab basis before applying the
- spectra_cblist callback functions
List of callback functions that compute the noise power spectrum as a function of frequency for the collapse operators in c_ops.
- Tfloat
The period of the time-dependence of the hamiltonian. The default value ‘None’ indicates that the ‘tlist’ spans a single period of the driving.
- w_thfloat
The temperature of the environment in units of frequency. For example, if the Hamiltonian written in units of 2pi GHz, and the temperature is given in K, use the following conversion:
temperature = 25e-3 # unit K h = 6.626e-34 kB = 1.38e-23 args[‘w_th’] = temperature * (kB / h) * 2 * pi * 1e-9
- argsdictionary
Dictionary of parameters for time-dependent Hamiltonian
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
store_floquet_states : bool Whether or not to store the density matrices in the floquet basis in
result.floquet_states
.normalize_output : bool Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’} How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str [“adams”, “bdf”, “lsoda”, “dop853”, “vern9”, etc.] Which differential equation integration method to use.
atol, rtol : float Absolute and relative tolerance of the ODE integrator.
nsteps : Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, 0 Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
Other options could be supported depending on the integration method, see Integrator.
- H
- Returns:
- result:
qutip.Result
An instance of the class
qutip.Result
, which contains the expectation values for the times specified by tlist, and/or the state density matrices corresponding to the times.
- result:
- fsesolve(H, psi0, tlist, e_ops=None, T=0.0, args=None, options=None)[source]
Solve the Schrodinger equation using the Floquet formalism.
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. Periodic system Hamiltonian as
QobjEvo
. List of [Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- psi0
qutip.qobj
Initial state vector (ket). If an operator is provided,
- tlistlist / array
List of times for \(t\).
- e_opslist of
qutip.qobj
/ callback function, optional List of operators for which to evaluate expectation values. If this list is empty, the state vectors for each time in tlist will be returned instead of expectation values.
- Tfloat, default=tlist[-1]
The period of the time-dependence of the hamiltonian.
- argsdictionary, optional
Dictionary with variables required to evaluate H.
- optionsdict, optional
Options for the results.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool Normalize output state to hide ODE numerical errors.
- H
- Returns:
- output
qutip.solver.Result
An instance of the class
qutip.solver.Result
, which contains either an array of expectation values or an array of state vectors, for the times specified by tlist.
- output
Stochastic Schrödinger Equation and Master Equation
- smesolve(H, rho0, tlist, c_ops=(), sc_ops=(), heterodyne=False, *, e_ops=(), args={}, ntraj=500, options=None, seeds=None, target_tol=None, timeout=None)[source]
Solve stochastic master equation.
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. System Hamiltonian as a Qobj or QobjEvo for time-dependent Hamiltonians. List of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- rho0
qutip.Qobj
Initial density matrix or state vector (ket).
- tlistlist / array
List of times for \(t\).
- c_opslist of (
QobjEvo
,QobjEvo
compatible format) Deterministic collapse operator which will contribute with a standard Lindblad type of dissipation.
- sc_opslist of (
QobjEvo
,QobjEvo
compatible format) List of stochastic collapse operators.
- e_ops:
qutip.qobj
, callable, or list. Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- argsNone / dictionary
Dictionary of parameters for time-dependent Hamiltonians and collapse operators.
- ntrajint [500]
Number of trajectories to compute.
- heterodynebool [False]
Whether to use heterodyne or homodyne detection.
- seedsint, SeedSequence, list, [optional]
Seed for the random number generator. It can be a single seed used to spawn seeds for each trajectory or a list of seeds, one for each trajectory. Seeds are saved in the result and they can be reused with:
seeds=prev_result.seeds
When using a parallel map, the trajectories can be re-ordered.
- target_tol{float, tuple, list}, optional
Target tolerance of the evolution. The evolution will compute trajectories until the error on the expectation values is lower than this tolerance. The maximum number of trajectories employed is given by
ntraj
. The error is computed using jackknife resampling.target_tol
can be an absolute tolerance or a pair of absolute and relative tolerance, in that order. Lastly, it can be a list of pairs of(atol, rtol)
for each e_ops.- timeoutfloat [optional]
Maximum time for the evolution in second. When reached, no more trajectories will be computed. Overwrite the option of the same name.
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool, [False] Whether or not to store the final state of the evolution in the result class.
store_states : bool, None, [None] Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
store_measurement: bool, [False] Whether to store the measurement and wiener process for each trajectories.
keep_runs_results : bool, [False] Whether to store results from all trajectories or just store the averages.
normalize_output : bool, [False] Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’}, [“text”] How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict, [{“chunk_size”: 10}] kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str, [“rouchon”] Which stochastic differential equation integration method to use. Main ones are {“euler”, “rouchon”, “platen”, “taylor1.5_imp”}
map : str {“serial”, “parallel”, “loky”}, [“serial”] How to run the trajectories. “parallel” uses concurent module to run in parallel while “loky” use the module of the same name to do so.
job_timeout : NoneType, int, [None] Maximum time to compute one trajectory.
num_cpus : NoneType, int, [None] Number of cpus to use when running in parallel.
None
detect the number of available cpus.dt : float [0.001 ~ 0.0001] The finite steps lenght for the Stochastic integration method. Default change depending on the integrator.
Other options could be supported depending on the integration method, see SIntegrator.
- H
- Returns:
- output:
qutip.solver.Result
An instance of the class
qutip.solver.Result
.
- output:
- ssesolve(H, psi0, tlist, sc_ops=(), heterodyne=False, *, e_ops=(), args={}, ntraj=500, options=None, seeds=None, target_tol=None, timeout=None)[source]
Solve stochastic Schrodinger equation.
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. System Hamiltonian as a Qobj or QobjEvo for time-dependent Hamiltonians. List of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- psi0
qutip.Qobj
Initial state vector (ket).
- tlistlist / array
List of times for \(t\).
- sc_opslist of (
QobjEvo
,QobjEvo
compatible format) List of stochastic collapse operators.
- e_ops
qutip.qobj
, callable, or list. Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- argsNone / dictionary
Dictionary of parameters for time-dependent Hamiltonians and collapse operators.
- ntrajint [500]
Number of trajectories to compute.
- heterodynebool [False]
Whether to use heterodyne or homodyne detection.
- seedsint, SeedSequence, list, [optional]
Seed for the random number generator. It can be a single seed used to spawn seeds for each trajectory or a list of seeds, one for each trajectory. Seeds are saved in the result and they can be reused with:
seeds=prev_result.seeds
- target_tol{float, tuple, list}, optional
Target tolerance of the evolution. The evolution will compute trajectories until the error on the expectation values is lower than this tolerance. The maximum number of trajectories employed is given by
ntraj
. The error is computed using jackknife resampling.target_tol
can be an absolute tolerance or a pair of absolute and relative tolerance, in that order. Lastly, it can be a list of pairs of (atol, rtol) for each e_ops.- timeoutfloat [optional]
Maximum time for the evolution in second. When reached, no more trajectories will be computed. Overwrite the option of the same name.
- optionsNone / dict
Dictionary of options for the solver.
store_final_state : bool, [False] Whether or not to store the final state of the evolution in the result class.
store_states : bool, None, [None] Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
store_measurement: bool, [False] Whether to store the measurement and wiener process, or brownian noise for each trajectories.
keep_runs_results : bool, [False] Whether to store results from all trajectories or just store the averages.
normalize_output : bool, [False] Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’}, [“text”] How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict, [{“chunk_size”: 10}] kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
method : str, [“rouchon”] Which stochastic differential equation integration method to use. Main ones are {“euler”, “rouchon”, “platen”, “taylor1.5_imp”}
map : str {“serial”, “parallel”, “loky”}, [“serial”] How to run the trajectories. “parallel” uses concurent module to run in parallel while “loky” use the module of the same name to do so.
job_timeout : NoneType, int, [None] Maximum time to compute one trajectory.
num_cpus : NoneType, int, [None] Number of cpus to use when running in parallel.
None
detect the number of available cpus.dt : float [0.001 ~ 0.0001] The finite steps lenght for the Stochastic integration method. Default change depending on the integrator.
Other options could be supported depending on the integration method, see SIntegrator.
- H
- Returns:
- output:
qutip.solver.Result
An instance of the class
qutip.solver.Result
.
- output:
Hierarchical Equations of Motion
This module provides solvers for system-bath evoluation using the HEOM (hierarchy equations of motion).
See https://en.wikipedia.org/wiki/Hierarchical_equations_of_motion for a very basic introduction to the technique.
The implementation is derived from the BoFiN library (see https://github.com/tehruhn/bofin) which was itself derived from an earlier implementation in QuTiP itself.
For backwards compatibility with QuTiP 4.6 and below, a new version of HSolverDL (the Drude-Lorentz specific HEOM solver) is provided. It is implemented on top of the new HEOMSolver but should largely be a drop-in replacement for the old HSolverDL.
- heomsolve(H, bath, max_depth, state0, tlist, *, e_ops=None, args=None, options=None)[source]
Hierarchical Equations of Motion (HEOM) solver that supports multiple baths.
The baths must be all either bosonic or fermionic baths.
If you need to run many evolutions of the same system and bath, consider using
HEOMSolver
directly to avoid having to continually reconstruct the equation hierarchy for every evolution.- Parameters:
- H
Qobj
,QobjEvo
Possibly time-dependent system Liouvillian or Hamiltonian as a Qobj or QobjEvo. list of [
Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- bathBath or list of Bath
A
Bath
containing the exponents of the expansion of the bath correlation funcion and their associated coefficients and coupling operators, or a list of baths.If multiple baths are given, they must all be either fermionic or bosonic baths.
- max_depthint
The maximum depth of the heirarchy (i.e. the maximum number of bath exponent “excitations” to retain).
- state0
Qobj
orHierarchyADOsState
or array-like If
rho0
is aQobj
the it is the initial state of the system (i.e. aQobj
density matrix).If it is a
HierarchyADOsState
or array-like, thenrho0
gives the initial state of all ADOs.Usually the state of the ADOs would be determine from a previous call to
.run(...)
with the solver results optionstore_ados
set to True. For example,result = solver.run(...)
could be followed bysolver.run(result.ado_states[-1], tlist)
.If a numpy array-like is passed its shape must be
(number_of_ados, n, n)
where(n, n)
is the system shape (i.e. shape of the system density matrix) and the ADOs must be in the same order as in.ados.labels
.- tlistlist
An ordered list of times at which to return the value of the state.
- e_opsQobj / QobjEvo / callable / list / dict / None, optional
A list or dictionary of operators as
Qobj
,QobjEvo
and/or callable functions (they can be mixed) or a single operator or callable function. For an operatorop
, the result will be computed using(state * op).tr()
and the state at each timet
. For callable functions,f
, the result is computed usingf(t, ado_state)
. The values are stored in theexpect
ande_data
attributes of the result (see the return section below).- argsdict, optional {None}
Change the
args
of the RHS for the evolution.- optionsdict, optional {None}
Generic solver options.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
store_ados : bool {False, True} Whether or not to store the HEOM ADOs.
normalize_output : bool Normalize output state to hide ODE numerical errors.
progress_bar : str {‘text’, ‘enhanced’, ‘tqdm’, ‘’} How to present the solver progress. ‘tqdm’ uses the python module of the same name and raise an error if not installed. Empty string or False will disable the bar.
progress_kwargs : dict kwargs to pass to the progress_bar. Qutip’s bars use chunk_size.
state_data_type: str {‘dense’} Name of the data type of the state used during the ODE evolution. Use an empty string to keep the input state type. Many integrator can only work with Dense.
method : str [“adams”, “bdf”, “lsoda”, “dop853”, “vern9”, etc.] Which differential equation integration method to use.
atol, rtol : float Absolute and relative tolerance of the ODE integrator.
nsteps : Maximum number of (internally defined) steps allowed in one
tlist
step.max_step : float, 0 Maximum lenght of one internal step. When using pulses, it should be less than half the width of the thinnest pulse.
- H
- Returns:
HEOMResult
The results of the simulation run, with the following important attributes:
times
: the timest
(i.e. thetlist
).states
: the system state at each timet
(only available ife_ops
wasNone
or if the solver optionstore_states
was set toTrue
).ado_states
: the full ADO state at each time (only available if the results optionado_return
was set toTrue
). Each element is an instance ofHierarchyADOsState
. The state of a particular ADO may be extracted fromresult.ado_states[i]
by callingextract
.expect
: a list containing the values of eache_ops
at timet
.e_data
: a dictionary containing the values of eache_ops
at tmet
. The keys are those given bye_ops
if it was a dict, otherwise they are the indexes of the suppliede_ops
.
See
HEOMResult
andResult
for the complete list of attributes.
Correlation Functions
- coherence_function_g1(H, state0, taulist, c_ops, a_op, solver='me', args={}, options={})[source]
Calculate the normalized first-order quantum coherence function:
\[g^{(1)}(\tau) = \frac{\langle A^\dagger(\tau)A(0)\rangle} {\sqrt{\langle A^\dagger(\tau)A(\tau)\rangle \langle A^\dagger(0)A(0)\rangle}}\]using the quantum regression theorem and the evolution solver indicated by the solver parameter.
- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist of {
Qobj
,QobjEvo
} List of collapse operators
- a_op
Qobj
,QobjEvo
Operator A.
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- g1, G1tuple
The normalized and unnormalized second-order coherence function.
- coherence_function_g2(H, state0, taulist, c_ops, a_op, solver='me', args={}, options={})[source]
Calculate the normalized second-order quantum coherence function:
\[ g^{(2)}(\tau) = \frac{\langle A^\dagger(0)A^\dagger(\tau)A(\tau)A(0)\rangle} {\langle A^\dagger(\tau)A(\tau)\rangle \langle A^\dagger(0)A(0)\rangle}\]using the quantum regression theorem and the evolution solver indicated by the solver parameter.
- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist
List of collapse operators, may be time-dependent for solver choice of me.
- a_op
Qobj
Operator A.
- argsdict
Dictionary of arguments to be passed to solver.
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- g2, G2tuple
The normalized and unnormalized second-order coherence function.
- correlation_2op_1t(H, state0, taulist, c_ops, a_op, b_op, solver='me', reverse=False, args={}, options={})[source]
Calculate the two-operator one-time correlation function: \(\left<A(\tau)B(0)\right>\) along one time axis using the quantum regression theorem and the evolution solver indicated by the solver parameter.
- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist of {
Qobj
,QobjEvo
} List of collapse operators
- a_op
Qobj
,QobjEvo
Operator A.
- b_op
Qobj
,QobjEvo
Operator B.
- reversebool {False}
If True, calculate \(\left<A(t)B(t+\tau)\right>\) instead of \(\left<A(t+\tau)B(t)\right>\).
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- corr_vecndarray
An array of correlation values for the times specified by taulist.
See also
correlation_3op
Similar function supporting various solver types.
References
See, Gardiner, Quantum Noise, Section 5.2.
- correlation_2op_2t(H, state0, tlist, taulist, c_ops, a_op, b_op, solver='me', reverse=False, args={}, options={})[source]
Calculate the two-operator two-time correlation function: \(\left<A(t+\tau)B(t)\right>\) along two time axes using the quantum regression theorem and the evolution solver indicated by the
solver
parameter.- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- tlistarray_like
List of times for \(t\). tlist must be positive and contain the element 0. When taking steady-steady correlations only one
tlist
value is necessary, i.e. when \(t \rightarrow \infty\). Iftlist
isNone
,tlist=[0]
is assumed.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist of {
Qobj
,QobjEvo
} List of collapse operators
- a_op
Qobj
,QobjEvo
Operator A.
- b_op
Qobj
,QobjEvo
Operator B.
- reversebool {False}
If True, calculate \(\left<A(t)B(t+\tau)\right>\) instead of \(\left<A(t+\tau)B(t)\right>\).
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- corr_matndarray
An 2-dimensional array (matrix) of correlation values for the times specified by tlist (first index) and taulist (second index).
See also
correlation_3op
Similar function supporting various solver types.
References
See, Gardiner, Quantum Noise, Section 5.2.
- correlation_3op(solver, state0, tlist, taulist, A=None, B=None, C=None)[source]
Calculate the three-operator two-time correlation function:
\(\left<A(t)B(t+\tau)C(t)\right>\).
from a open system
Solver
.Note: it is not possible to calculate a physically meaningful correlation where \(\tau<0\).
- Parameters:
- solver
MESolver
,BRSolver
Qutip solver for an open system.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\).
- tlistarray_like
List of times for \(t\). tlist must be positive and contain the element 0.
- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- A, B, C: :class:`Qobj`, :class:`QobjEvo`, optional, default=None
Operators
A
,B
,C
from the equation<A(t)B(t+\tau)C(t)>
in the Schrodinger picture. They do not need to be all provided. For exemple, ifA
is not provided,<B(t+\tau)C(t)>
is computed.
- solver
- Returns:
- corr_matarray
An 2-dimensional array (matrix) of correlation values for the times specified by tlist (first index) and taulist (second index). If tlist is None, then a 1-dimensional array of correlation values is returned instead.
- correlation_3op_1t(H, state0, taulist, c_ops, a_op, b_op, c_op, solver='me', args={}, options={})[source]
Calculate the three-operator two-time correlation function: \(\left<A(0)B(\tau)C(0)\right>\) along one time axis using the quantum regression theorem and the evolution solver indicated by the solver parameter.
Note: it is not possibly to calculate a physically meaningful correlation of this form where \(\tau<0\).
- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist of {
Qobj
,QobjEvo
} List of collapse operators
- a_op
Qobj
,QobjEvo
Operator A.
- b_op
Qobj
,QobjEvo
Operator B.
- c_op
Qobj
,QobjEvo
Operator C.
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- corr_vecarray
An array of correlation values for the times specified by taulist.
See also
correlation_3op
Similar function supporting various solver types.
References
See, Gardiner, Quantum Noise, Section 5.2.
- correlation_3op_2t(H, state0, tlist, taulist, c_ops, a_op, b_op, c_op, solver='me', args={}, options={})[source]
Calculate the three-operator two-time correlation function: \(\left<A(t)B(t+\tau)C(t)\right>\) along two time axes using the quantum regression theorem and the evolution solver indicated by the solver parameter.
Note: it is not possibly to calculate a physically meaningful correlation of this form where \(\tau<0\).
- Parameters:
- H
Qobj
,QobjEvo
System Hamiltonian, may be time-dependent for solver choice of me.
- state0
Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\). If ‘state0’ is ‘None’, then the steady state will be used as the initial state. The ‘steady-state’ is only implemented if
c_ops
are provided and the Hamiltonian is constant.- tlistarray_like
List of times for \(t\). tlist must be positive and contain the element 0. When taking steady-steady correlations only one tlist value is necessary, i.e. when \(t \rightarrow \infty\). If
tlist
isNone
,tlist=[0]
is assumed.- taulistarray_like
List of times for \(\tau\). taulist must be positive and contain the element 0.
- c_opslist of {
Qobj
,QobjEvo
} List of collapse operators
- a_op
Qobj
,QobjEvo
Operator A.
- b_op
Qobj
,QobjEvo
Operator B.
- c_op
Qobj
,QobjEvo
Operator C.
- solverstr {‘me’, ‘es’}
Choice of solver, me for master-equation, and es for exponential series. es is equivalent to me with
options={"method": "diag"}
.- optionsdict, optional
Options for the solver. Only used with me solver.
- H
- Returns:
- corr_matarray
An 2-dimensional array (matrix) of correlation values for the times specified by tlist (first index) and taulist (second index).
See also
correlation_3op
Similar function supporting various solver types.
References
See, Gardiner, Quantum Noise, Section 5.2.
- spectrum(H, wlist, c_ops, a_op, b_op, solver='es')[source]
Calculate the spectrum of the correlation function \(\lim_{t \to \infty} \left<A(t+\tau)B(t)\right>\), i.e., the Fourier transform of the correlation function:
\[S(\omega) = \int_{-\infty}^{\infty} \lim_{t \to \infty} \left<A(t+\tau)B(t)\right> e^{-i\omega\tau} d\tau.\]using the solver indicated by the solver parameter. Note: this spectrum is only defined for stationary statistics (uses steady state rho0)
- Parameters:
- H
qutip.qobj
system Hamiltonian.
- wlistarray_like
List of frequencies for \(\omega\).
- c_opslist
List of collapse operators.
- a_opQobj
Operator A.
- b_opQobj
Operator B.
- solverstr
Choice of solver (es for exponential series and pi for psuedo-inverse, solve for generic solver).
- H
- Returns:
- spectrumarray
An array with spectrum \(S(\omega)\) for the frequencies specified in wlist.
- spectrum_correlation_fft(tlist, y, inverse=False)[source]
Calculate the power spectrum corresponding to a two-time correlation function using FFT.
- Parameters:
- tlistarray_like
list/array of times \(t\) which the correlation function is given.
- yarray_like
list/array of correlations corresponding to time delays \(t\).
- inverse: boolean
boolean parameter for using a positive exponent in the Fourier Transform instead. Default is False.
- Returns:
- w, Stuple
Returns an array of angular frequencies ‘w’ and the corresponding two-sided power spectrum ‘S(w)’.
Steady-state Solvers
- pseudo_inverse(L, rhoss=None, w=None, method='splu', *, use_rcm=False, **kwargs)[source]
Compute the pseudo inverse for a Liouvillian superoperator, optionally given its steady state density matrix (which will be computed if not given).
- Parameters:
- LQobj
A Liouvillian superoperator for which to compute the pseudo inverse.
- rhossQobj
A steadystate density matrix as Qobj instance, for the Liouvillian superoperator L.
- wdouble
frequency at which to evaluate pseudo-inverse. Can be zero for dense systems and large sparse systems. Small sparse systems can fail for zero frequencies.
- sparsebool
Flag that indicate whether to use sparse or dense matrix methods when computing the pseudo inverse.
- methodstring
Method used to compte matrix inverse. Choice are ‘pinv’ to use scipy’s function of the same name, or a linear system solver. Default supported solver are:
“solve”, “lstsq” dense solver from numpy.linalg
“spsolve”, “gmres”, “lgmres”, “bicgstab”, “splu” sparse solver from scipy.sparse.linalg
“mkl_spsolve”, sparse solver by mkl.
Extension to qutip, such as qutip-tensorflow, can use come with their own solver. When
L
use these data backends, see the corresponding librarieslinalg
for available solver.- kwargsdictionary
Additional keyword arguments for setting parameters for solver methods.
- Returns:
- RQobj
Returns a Qobj instance representing the pseudo inverse of L.
Note
In general the inverse of a sparse matrix will be dense. If you are applying the inverse to a density matrix then it is better to cast the problem as an Ax=b type problem where the explicit calculation of the inverse is not required. See page 67 of “Electrons in nanostructures” C. Flindt, PhD Thesis available online: https://orbit.dtu.dk/fedora/objects/orbit:82314/datastreams/ file_4732600/content
Note also that the definition of the pseudo-inverse herein is different from numpys pinv() alone, as it includes pre and post projection onto the subspace defined by the projector Q.
- steadystate(A, c_ops=[], *, method='direct', solver=None, **kwargs)[source]
Calculates the steady state for quantum evolution subject to the supplied Hamiltonian or Liouvillian operator and (if given a Hamiltonian) a list of collapse operators.
If the user passes a Hamiltonian then it, along with the list of collapse operators, will be converted into a Liouvillian operator in Lindblad form.
- Parameters:
- A
Qobj
A Hamiltonian or Liouvillian operator.
- c_op_listlist
A list of collapse operators.
- methodstr, default=’direct’
The allowed methods are composed of 2 parts, the steadystate method: - “direct”: Solving
L(rho_ss) = 0
- “eigen” : Eigenvalue problem - “svd” : Singular value decomposition - “power” : Inverse-power method- solverstr, default=None
‘direct’ and ‘power’ methods only. Solver to use when solving the
L(rho_ss) = 0
equation. Default supported solver are:“solve”, “lstsq” dense solver from numpy.linalg
“spsolve”, “gmres”, “lgmres”, “bicgstab” sparse solver from scipy.sparse.linalg
“mkl_spsolve” sparse solver by mkl.
Extension to qutip, such as qutip-tensorflow, can use come with their own solver. When
A
andc_ops
use these data backends, see the corresponding librarieslinalg
for available solver.Extra options for these solver can be passed in
**kw
.- use_rcmbool, default False
Use reverse Cuthill-Mckee reordering to minimize fill-in in the LU factorization of the Liouvillian. Used with ‘direct’ or ‘power’ method.
- use_wbmbool, default False
Use Weighted Bipartite Matching reordering to make the Liouvillian diagonally dominant. This is useful for iterative preconditioners only. Used with ‘direct’ or ‘power’ method.
- weightfloat, optional
Sets the size of the elements used for adding the unity trace condition to the linear solvers. This is set to the average abs value of the Liouvillian elements if not specified by the user. Used with ‘direct’ method.
- power_tolfloat, default 1e-12
Tolerance for the solution when using the ‘power’ method.
- power_maxiterint, default 10
Maximum number of iteration to use when looking for a solution when using the ‘power’ method.
- power_eps: double, default 1e-15
Small weight used in the “power” method.
- sparse: bool
Whether to use the sparse eigen solver with the “eigen” method (default sparse). With “direct” and “power” method, when the solver is not specified, it is used to set whether “solve” or “spsolve” is used as default solver.
- **kwargs
Extra options to pass to the linear system solver. See the documentation of the used solver in
numpy.linalg
orscipy.sparse.linalg
to see what extra arguments are supported.
- A
- Returns:
- dmqobj
Steady state density matrix.
- infodict, optional
Dictionary containing solver-specific information about the solution.
Note
The SVD method works only for dense operators (i.e. small systems).
- steadystate_floquet(H_0, c_ops, Op_t, w_d=1.0, n_it=3, sparse=False, solver=None, **kwargs)[source]
- Calculates the effective steady state for a driven
system with a time-dependent cosinusoidal term:
\[\mathcal{\hat{H}}(t) = \hat{H}_0 + \mathcal{\hat{O}} \cos(\omega_d t)\]- Parameters:
- H_0
Qobj
A Hamiltonian or Liouvillian operator.
- c_opslist
A list of collapse operators.
- Op_t
Qobj
The the interaction operator which is multiplied by the cosine
- w_dfloat, default 1.0
The frequency of the drive
- n_itint, default 3
The number of iterations for the solver
- sparsebool, default False
Solve for the steady state using sparse algorithms.
- solverstr, default=None
Solver to use when solving the linear system. Default supported solver are:
“solve”, “lstsq” dense solver from numpy.linalg
“spsolve”, “gmres”, “lgmres”, “bicgstab” sparse solver from scipy.sparse.linalg
“mkl_spsolve” sparse solver by mkl.
Extensions to qutip, such as qutip-tensorflow, may provide their own solvers. When
H_0
andc_ops
use these data backends, see their documentation for the names and details of additional solvers they may provide.- **kwargs:
Extra options to pass to the linear system solver. See the documentation of the used solver in
numpy.linalg
orscipy.sparse.linalg
to see what extra arguments are supported.
- H_0
- Returns:
- dmqobj
Steady state density matrix.
Note
See: Sze Meng Tan, https://copilot.caltech.edu/documents/16743/qousersguide.pdf, Section (10.16)
Propagators
- propagator(H, t, c_ops=(), args=None, options=None, **kwargs)[source]
Calculate the propagator U(t) for the density matrix or wave function such that \(\psi(t) = U(t)\psi(0)\) or \(\rho_{\mathrm vec}(t) = U(t) \rho_{\mathrm vec}(0)\) where \(\rho_{\mathrm vec}\) is the vector representation of the density matrix.
- Parameters:
- H
Qobj
,QobjEvo
,QobjEvo
compatible format. Possibly time-dependent system Liouvillian or Hamiltonian as a Qobj or QobjEvo.
list
of [Qobj
,Coefficient
] or callable that can be made intoQobjEvo
are also accepted.- tfloat or array-like
Time or list of times for which to evaluate the propagator.
- c_opslist, optional
List of Qobj or QobjEvo collapse operators.
- argsdictionary, optional
Parameters to callback functions for time-dependent Hamiltonians and collapse operators.
- optionsdict, optional
Options for the solver.
- **kwargs
Extra parameters to use when creating the
QobjEvo
from a list formatH
.
- H
- Returns:
- Uqobj, list
Instance representing the propagator(s) \(U(t)\). Return a single Qobj when
t
is a number or a list whent
is a list.
Scattering in Quantum Optical Systems
Photon scattering in quantum optical systems
This module includes a collection of functions for numerically computing photon scattering in driven arbitrary systems coupled to some configuration of output waveguides. The implementation of these functions closely follows the mathematical treatment given in K.A. Fischer, et. al., Scattering of Coherent Pulses from Quantum Optical Systems (2017, arXiv:1710.02875).
- scattering_probability(H, psi0, n_emissions, c_ops, tlist, system_zero_state=None, construct_effective_hamiltonian=True)[source]
Compute the integrated probability of scattering n photons in an arbitrary system. This function accepts a nonlinearly spaced array of times.
- Parameters:
- H:class: qutip.Qobj or list
System-waveguide(s) Hamiltonian or effective Hamiltonian in Qobj or list-callback format. If construct_effective_hamiltonian is not specified, an effective Hamiltonian is constructed from H and c_ops.
- psi0:class: qutip.Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\).
- n_emissionsint
Number of photons emitted by the system (into any combination of waveguides).
- c_opslist
List of collapse operators for each waveguide; these are assumed to include spontaneous decay rates, e.g. \(\sigma = \sqrt \gamma \cdot a\).
- tlistarray_like
List of times for \(\tau_i\). tlist should contain 0 and exceed the pulse duration / temporal region of interest; tlist need not be linearly spaced.
- system_zero_state:class: qutip.Qobj
State representing zero excitations in the system. Defaults to basis(systemDims, 0).
- construct_effective_hamiltonianbool
Whether an effective Hamiltonian should be constructed from H and c_ops: \(H_{eff} = H - \frac{i}{2} \sum_n \sigma_n^\dagger \sigma_n\) Default: True.
- Returns:
- scattering_probfloat
The probability of scattering n photons from the system over the time range specified.
- temporal_basis_vector(waveguide_emission_indices, n_time_bins, n_emissions=None)[source]
Generate a temporal basis vector for emissions at specified time bins into specified waveguides.
- Parameters:
- waveguide_emission_indiceslist or tuple
List of indices where photon emission occurs for each waveguide, e.g. [[t1_wg1], [t1_wg2, t2_wg2], [], [t1_wg4, t2_wg4, t3_wg4]].
- n_time_binsint
Number of time bins; the range over which each index can vary.
- Returns:
- temporal_basis_vector:class: qutip.Qobj
A basis vector representing photon scattering at the specified indices. If there are W waveguides, T times, and N photon emissions, then the basis vector has dimensionality (W*T)^N.
- temporal_scattered_state(H, psi0, n_emissions, c_ops, tlist, system_zero_state=None, construct_effective_hamiltonian=True)[source]
Compute the scattered n-photon state projected onto the temporal basis.
- Parameters:
- H:class: qutip.Qobj or list
System-waveguide(s) Hamiltonian or effective Hamiltonian in Qobj or list-callback format. If construct_effective_hamiltonian is not specified, an effective Hamiltonian is constructed from H and c_ops.
- psi0:class: qutip.Qobj
Initial state density matrix \(\rho(t_0)\) or state vector \(\psi(t_0)\).
- n_emissionsint
Number of photon emissions to calculate.
- c_opslist
List of collapse operators for each waveguide; these are assumed to include spontaneous decay rates, e.g. \(\sigma = \sqrt \gamma \cdot a\)
- tlistarray_like
List of times for \(\tau_i\). tlist should contain 0 and exceed the pulse duration / temporal region of interest.
- system_zero_state:class: qutip.Qobj
State representing zero excitations in the system. Defaults to \(\psi(t_0)\)
- construct_effective_hamiltonianbool
Whether an effective Hamiltonian should be constructed from H and c_ops: \(H_{eff} = H - \frac{i}{2} \sum_n \sigma_n^\dagger \sigma_n\) Default: True.
- Returns:
- phi_n:class: qutip.Qobj
The scattered bath state projected onto the temporal basis given by tlist. If there are W waveguides, T times, and N photon emissions, then the state is a tensor product state with dimensionality T^(W*N).
Permutational Invariance
Permutational Invariant Quantum Solver (PIQS)
This module calculates the Liouvillian for the dynamics of ensembles of identical two-level systems (TLS) in the presence of local and collective processes by exploiting permutational symmetry and using the Dicke basis. It also allows to characterize nonlinear functions of the density matrix.
- am(j, m)[source]
Calculate the operator
am
used later.The action of
ap
is given by: \(J_{-}\lvert j,m\rangle = A_{-}(jm)\lvert j,m-1\rangle\)- Parameters:
- j: float
The value for j.
- m: float
The value for m.
- Returns:
- a_minus: float
The value of \(a_{-}\).
- ap(j, m)[source]
Calculate the coefficient
ap
by applying \(J_+\lvert j,m\rangle\).The action of ap is given by: \(J_{+}\lvert j, m\rangle = A_{+}(j, m) \lvert j, m+1\rangle\)
- Parameters:
- j, m: float
The value for j and m in the dicke basis \(\lvert j, m\rangle\).
- Returns:
- a_plus: float
The value of \(a_{+}\).
- block_matrix(N, elements='ones')[source]
Construct the block-diagonal matrix for the Dicke basis.
- Parameters:
- Nint
Number of two-level systems.
- elementsstr {‘ones’ (default),’degeneracy’}
- Returns:
- block_matrndarray
A 2D block-diagonal matrix with dimension (nds,nds), where nds is the number of Dicke states for N two-level systems. Filled with ones or the value of degeneracy at each matrix element.
- collapse_uncoupled(N, emission=0.0, dephasing=0.0, pumping=0.0, collective_emission=0.0, collective_dephasing=0.0, collective_pumping=0.0)[source]
Create the collapse operators (c_ops) of the Lindbladian in the uncoupled basis
These operators are in the uncoupled basis of the two-level system (TLS) SU(2) Pauli matrices.
- Parameters:
- N: int
The number of two-level systems.
- emission: float
Incoherent emission coefficient (also nonradiative emission). default: 0.0
- dephasing: float
Local dephasing coefficient. default: 0.0
- pumping: float
Incoherent pumping coefficient. default: 0.0
- collective_emission: float
Collective (superradiant) emmission coefficient. default: 0.0
- collective_pumping: float
Collective pumping coefficient. default: 0.0
- collective_dephasing: float
Collective dephasing coefficient. default: 0.0
- Returns:
- c_ops: list
The list of collapse operators as qutip.Qobj for the system.
Notes
The collapse operator list can be given to qutip.mesolve. Notice that the operators are placed in a Hilbert space of dimension \(2^N\). Thus the method is suitable only for small N (of the order of 10).
- css(N, x=0.7071067811865475, y=0.7071067811865475, basis='dicke', coordinates='cartesian')[source]
Generate the density matrix of the Coherent Spin State (CSS).
It can be defined as, \(\lvert CSS\rangle = \prod_i^N(a\lvert1\rangle_i+b\lvert0\rangle_i)\) with \(a = sin(\frac{\theta}{2})\), \(b = e^{i \phi}\cos(\frac{\theta}{2})\). The default basis is that of Dicke space \(\lvert j, m\rangle \langle j, m'\rvert\). The default state is the symmetric CSS, \(\lvert CSS\rangle = \lvert+\rangle\).
- Parameters:
- N: int
The number of two-level systems.
- x, y: float
The coefficients of the CSS state.
- basis: str
The basis to use. Either “dicke” or “uncoupled”.
- coordinates: str
Either “cartesian” or “polar”. If polar then the coefficients are constructed as sin(x/2), cos(x/2)e^(iy).
- Returns:
- rho: :class: qutip.Qobj
The CSS state density matrix.
- dicke(N, j, m)[source]
Generate a Dicke state as a pure density matrix in the Dicke basis.
For instance, the superradiant state given by \(\lvert j, m\rangle = \lvert 1, 0\rangle\) for N = 2, and the state is represented as a density matrix of size (nds, nds) or (4, 4), with the (1, 1) element set to 1.
- Parameters:
- N: int
The number of two-level systems.
- j: float
The eigenvalue j of the Dicke state (j, m).
- m: float
The eigenvalue m of the Dicke state (j, m).
- Returns:
- rho: :class: qutip.Qobj
The density matrix.
- dicke_basis(N, jmm1=None)[source]
Initialize the density matrix of a Dicke state for several (j, m, m1).
This function can be used to build arbitrary states in the Dicke basis \(\lvert j, m\rangle\langle j, m'\rvert\). We create coefficients for each (j, m, m1) value in the dictionary jmm1. The mapping for the (i, k) index of the density matrix to the \(\lvert j, m\rangle\) values is given by the cythonized function jmm1_dictionary. A density matrix is created from the given dictionary of coefficients for each (j, m, m1).
- Parameters:
- N: int
The number of two-level systems.
- jmm1: dict
A dictionary of {(j, m, m1): p} that gives a density p for the (j, m, m1) matrix element.
- Returns:
- rho: :class: qutip.Qobj
The density matrix in the Dicke basis.
- dicke_blocks(rho)[source]
Create the list of blocks for block-diagonal density matrix in the Dicke basis.
- Parameters:
- rho
qutip.Qobj
A 2D block-diagonal matrix of ones with dimension (nds,nds), where nds is the number of Dicke states for N two-level systems.
- rho
- Returns:
- square_blocks: list of np.array
Give back the blocks list.
- dicke_blocks_full(rho)[source]
Give the full (2^N-dimensional) list of blocks for a Dicke-basis matrix.
- Parameters:
- rho
qutip.Qobj
A 2D block-diagonal matrix of ones with dimension (nds,nds), where nds is the number of Dicke states for N two-level systems.
- rho
- Returns:
- full_blockslist
The list of blocks expanded in the 2^N space for N qubits.
- dicke_function_trace(f, rho)[source]
Calculate the trace of a function on a Dicke density matrix. :param f: A Taylor-expandable function of rho. :type f: function :param rho: A density matrix in the Dicke basis. :type rho:
qutip.Qobj
- Returns:
- resfloat
Trace of a nonlinear function on rho.
- energy_degeneracy(N, m)[source]
Calculate the number of Dicke states with same energy.
The use of the Decimals class allows to explore N > 1000, unlike the built-in function scipy.special.binom
- Parameters:
- N: int
The number of two-level systems.
- m: float
Total spin z-axis projection eigenvalue. This is proportional to the total energy.
- Returns:
- degeneracy: int
The energy degeneracy
- entropy_vn_dicke(rho)[source]
Von Neumann Entropy of a Dicke-basis density matrix.
- Parameters:
- rho
qutip.Qobj
A 2D block-diagonal matrix of ones with dimension (nds,nds), where nds is the number of Dicke states for N two-level systems.
- rho
- Returns:
- entropy_dm: float
Entropy. Use degeneracy to multiply each block.
- excited(N, basis='dicke')[source]
Generate the density matrix for the excited state.
This state is given by (N/2, N/2) in the default Dicke basis. If the argument basis is “uncoupled” then it generates the state in a 2**N dim Hilbert space.
- Parameters:
- N: int
The number of two-level systems.
- basis: str
The basis to use. Either “dicke” or “uncoupled”.
- Returns:
- state: :class: qutip.Qobj
The excited state density matrix in the requested basis.
- ghz(N, basis='dicke')[source]
Generate the density matrix of the GHZ state.
If the argument basis is “uncoupled” then it generates the state in a \(2^N\)-dimensional Hilbert space.
- Parameters:
- N: int
The number of two-level systems.
- basis: str
The basis to use. Either “dicke” or “uncoupled”.
- Returns:
- state: :class: qutip.Qobj
The GHZ state density matrix in the requested basis.
- ground(N, basis='dicke')[source]
Generate the density matrix of the ground state.
This state is given by (N/2, -N/2) in the Dicke basis. If the argument basis is “uncoupled” then it generates the state in a \(2^N\)-dimensional Hilbert space.
- Parameters:
- N: int
The number of two-level systems.
- basis: str
The basis to use. Either “dicke” or “uncoupled”
- Returns:
- state: :class: qutip.Qobj
The ground state density matrix in the requested basis.
- identity_uncoupled(N)[source]
Generate the identity in a \(2^N\)-dimensional Hilbert space.
The identity matrix is formed from the tensor product of N TLSs.
- Parameters:
- N: int
The number of two-level systems.
- Returns:
- identity: :class: qutip.Qobj
The identity matrix.
- isdiagonal(mat)[source]
Check if the input matrix is diagonal.
- Parameters:
- mat: ndarray/Qobj
A 2D numpy array
- Returns:
- diag: bool
True/False depending on whether the input matrix is diagonal.
- jspin(N, op=None, basis='dicke')[source]
Calculate the list of collective operators of the total algebra.
The Dicke basis \(\lvert j,m\rangle\langle j,m'\rvert\) is used by default. Otherwise with “uncoupled” the operators are in a \(2^N\) space.
- Parameters:
- N: int
Number of two-level systems.
- op: str
The operator to return ‘x’,’y’,’z’,’+’,’-‘. If no operator given, then output is the list of operators for [‘x’,’y’,’z’].
- basis: str
The basis of the operators - “dicke” or “uncoupled” default: “dicke”.
- Returns:
- j_alg: list or :class: qutip.Qobj
A list of qutip.Qobj representing all the operators in the “dicke” or “uncoupled” basis or a single operator requested.
- m_degeneracy(N, m)[source]
Calculate the number of Dicke states \(\lvert j, m\rangle\) with same energy.
- Parameters:
- N: int
The number of two-level systems.
- m: float
Total spin z-axis projection eigenvalue (proportional to the total energy).
- Returns:
- degeneracy: int
The m-degeneracy.
- num_dicke_ladders(N)[source]
Calculate the total number of ladders in the Dicke space.
For a collection of N two-level systems it counts how many different “j” exist or the number of blocks in the block-diagonal matrix.
- Parameters:
- N: int
The number of two-level systems.
- Returns:
- Nj: int
The number of Dicke ladders.
- num_dicke_states(N)[source]
Calculate the number of Dicke states.
- Parameters:
- N: int
The number of two-level systems.
- Returns:
- nds: int
The number of Dicke states.
- num_tls(nds)[source]
Calculate the number of two-level systems.
- Parameters:
- nds: int
The number of Dicke states.
- Returns:
- N: int
The number of two-level systems.
- purity_dicke(rho)[source]
Calculate purity of a density matrix in the Dicke basis. It accounts for the degenerate blocks in the density matrix.
- Parameters:
- rho
qutip.Qobj
Density matrix in the Dicke basis of qutip.piqs.jspin(N), for N spins.
- rho
- Returns:
- purityfloat
The purity of the quantum state. It’s 1 for pure states, 0<=purity<1 for mixed states.
- spin_algebra(N, op=None)[source]
Create the list [sx, sy, sz] with the spin operators.
The operators are constructed for a collection of N two-level systems (TLSs). Each element of the list, i.e., sx, is a vector of qutip.Qobj objects (spin matrices), as it cointains the list of the SU(2) Pauli matrices for the N TLSs. Each TLS operator sx[i], with i = 0, …, (N-1), is placed in a \(2^N\)-dimensional Hilbert space.
- Parameters:
- N: int
The number of two-level systems.
- Returns:
- spin_operators: list or :class: qutip.Qobj
A list of qutip.Qobj operators - [sx, sy, sz] or the requested operator.
Notes
sx[i] is \(\frac{\sigma_x}{2}\) in the composite Hilbert space.
- state_degeneracy(N, j)[source]
Calculate the degeneracy of the Dicke state.
Each state \(\lvert j, m\rangle\) includes D(N,j) irreducible representations \(\lvert j, m, \alpha\rangle\).
Uses Decimals to calculate higher numerator and denominators numbers.
- Parameters:
- N: int
The number of two-level systems.
- j: float
Total spin eigenvalue (cooperativity).
- Returns:
- degeneracy: int
The state degeneracy.
- superradiant(N, basis='dicke')[source]
Generate the density matrix of the superradiant state.
This state is given by (N/2, 0) or (N/2, 0.5) in the Dicke basis. If the argument basis is “uncoupled” then it generates the state in a 2**N dim Hilbert space.
- Parameters:
- N: int
The number of two-level systems.
- basis: str
The basis to use. Either “dicke” or “uncoupled”.
- Returns:
- state: :class: qutip.Qobj
The superradiant state density matrix in the requested basis.
- tau_column(tau, k, j)[source]
Determine the column index for the non-zero elements of the matrix for a particular row k and the value of j from the Dicke space.
- Parameters:
- tau: str
The tau function to check for this k and j.
- k: int
The row of the matrix M for which the non zero elements have to be calculated.
- j: float
The value of j for this row.
Visualization
Pseudoprobability Functions
- qfunc(state: Qobj, xvec, yvec, g: float = 1.4142135623730951, precompute_memory: float = 1024)[source]
Husimi-Q function of a given state vector or density matrix at phase-space points
0.5 * g * (xvec + i*yvec)
.- Parameters:
- state
Qobj
A state vector or density matrix. This cannot have tensor-product structure.
- xvec, yvecarray_like
x- and y-coordinates at which to calculate the Husimi-Q function.
- gfloat, default sqrt(2)
Scaling factor for
a = 0.5 * g * (x + iy)
. The value of g is related to the value of \(\hbar\) in the commutation relation \([x,\,y] = i\hbar\) via \(\hbar=2/g^2\), so the default corresponds to \(\hbar=1\).- precompute_memoryreal, default 1024
Size in MB that may be used during calculations as working space when dealing with density-matrix inputs. This is ignored for state-vector inputs. The bound is not quite exact due to other, order-of-magnitude smaller, intermediaries being necessary, but is a good approximation. If you want to use the same iterative algorithm for density matrices that is used for single kets, set
precompute_memory=None
.
- state
- Returns:
- ndarray
Values representing the Husimi-Q function calculated over the specified range
[xvec, yvec]
.
See also
QFunc
a class-based version, more efficient if you want to calculate the Husimi-Q function for several states over the same coordinates.
- spin_q_function(rho, theta, phi)[source]
The Husimi Q function for spins is defined as
Q(theta, phi) = SCS.dag() * rho * SCS
for the spin coherent stateSCS = spin_coherent( j, theta, phi)
where j is the spin length. The implementation here is more efficient as it doesn’t generate all of the SCS at theta and phi (see references).The spin Q function is normal when integrated over the surface of the sphere
\[\frac{4 \pi}{2j + 1}\int_\phi \int_\theta Q(\theta, \phi) \sin(\theta) d\theta d\phi = 1\]- Parameters:
- stateqobj
A state vector or density matrix for a spin-j quantum system.
- thetaarray_like
Polar (colatitude) angle at which to calculate the Husimi-Q function.
- phiarray_like
Azimuthal angle at which to calculate the Husimi-Q function.
- Returns:
- Q, THETA, PHI2d-array
Values representing the spin Husimi Q function at the values specified by THETA and PHI.
References
[1] Lee Loh, Y., & Kim, M. (2015). American J. of Phys., 83(1), 30–35. https://doi.org/10.1119/1.4898595
- spin_wigner(rho, theta, phi)[source]
Wigner function for a spin-j system.
The spin W function is normal when integrated over the surface of the sphere
\[\sqrt{\frac{4 \pi}{2j + 1}}\int_\phi \int_\theta W(\theta,\phi) \sin(\theta) d\theta d\phi = 1\]- Parameters:
- stateqobj
A state vector or density matrix for a spin-j quantum system.
- thetaarray_like
Polar (colatitude) angle at which to calculate the W function.
- phiarray_like
Azimuthal angle at which to calculate the W function.
- Returns:
- W, THETA, PHI2d-array
Values representing the spin Wigner function at the values specified by THETA and PHI.
References
[1] Agarwal, G. S. (1981). Phys. Rev. A, 24(6), 2889–2896. https://doi.org/10.1103/PhysRevA.24.2889
[2] Dowling, J. P., Agarwal, G. S., & Schleich, W. P. (1994). Phys. Rev. A, 49(5), 4101–4109. https://doi.org/10.1103/PhysRevA.49.4101
[3] Conversion between Wigner 3-j symbol and Clebsch-Gordan coefficients taken from Wikipedia (https://en.wikipedia.org/wiki/3-j_symbol)
- wigner(psi, xvec, yvec, method='clenshaw', g=1.4142135623730951, sparse=False, parfor=False)[source]
Wigner function for a state vector or density matrix at points xvec + i * yvec.
- Parameters:
- stateqobj
A state vector or density matrix.
- xvecarray_like
x-coordinates at which to calculate the Wigner function.
- yvecarray_like
y-coordinates at which to calculate the Wigner function. Does not apply to the ‘fft’ method.
- gfloat
Scaling factor for a = 0.5 * g * (x + iy), default g = sqrt(2). The value of g is related to the value of hbar in the commutation relation [x, y] = i * hbar via hbar=2/g^2 giving the default value hbar=1.
- methodstring {‘clenshaw’, ‘iterative’, ‘laguerre’, ‘fft’}
Select method ‘clenshaw’ ‘iterative’, ‘laguerre’, or ‘fft’, where ‘clenshaw’ and ‘iterative’ use an iterative method to evaluate the Wigner functions for density matrices \(|m><n|\), while ‘laguerre’ uses the Laguerre polynomials in scipy for the same task. The ‘fft’ method evaluates the Fourier transform of the density matrix. The ‘iterative’ method is default, and in general recommended, but the ‘laguerre’ method is more efficient for very sparse density matrices (e.g., superpositions of Fock states in a large Hilbert space). The ‘clenshaw’ method is the preferred method for dealing with density matrices that have a large number of excitations (>~50). ‘clenshaw’ is a fast and numerically stable method.
- sparsebool {False, True}
Tells the default solver whether or not to keep the input density matrix in sparse format. As the dimensions of the density matrix grow, setthing this flag can result in increased performance.
- parforbool {False, True}
Flag for calculating the Laguerre polynomial based Wigner function method=’laguerre’ in parallel using the parfor function.
- Returns:
- Warray
Values representing the Wigner function calculated over the specified range [xvec,yvec].
- yvexarray
FFT ONLY. Returns the y-coordinate values calculated via the Fourier transform.
Notes
The ‘fft’ method accepts only an xvec input for the x-coordinate. The y-coordinates are calculated internally.
References
Ulf Leonhardt, Measuring the Quantum State of Light, (Cambridge University Press, 1997)
Graphs and Visualization
Functions for visualizing results of quantum dynamics simulations, visualizations of quantum states and processes.
- hinton(rho, x_basis=None, y_basis=None, color_style='scaled', label_top=True, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Draws a Hinton diagram to visualize a density matrix or superoperator.
- Parameters:
- rhoqobj
Input density matrix or superoperator.
Note
Hinton plots of superoperators are currently only supported for qubits.
- x_basislist of strings, optional
list of x ticklabels to represent x basis of the input.
- y_basislist of strings, optional
list of y ticklabels to represent y basis of the input.
- color_stylestring, default=”scaled”
Determines how colors are assigned to each square:
If set to
"scaled"
(default), each color is chosen by passing the absolute value of the corresponding matrix element into cmap with the sign of the real part.If set to
"threshold"
, each square is plotted as the maximum of cmap for the positive real part and as the minimum for the negative part of the matrix element; note that this generalizes “threshold” to complex numbers.If set to
"phase"
, each color is chosen according to the angle of the corresponding matrix element.
- label_topbool, default=True
If True, x ticklabels will be placed on top, otherwise they will appear below the plot.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The ax context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- Raises:
- ValueError
Input argument is not a quantum object.
Examples
>>> import qutip >>> >>> dm = qutip.rand_dm(4) >>> fig, ax = qutip.hinton(dm) >>> fig.show() >>> >>> qutip.settings.colorblind_safe = True >>> fig, ax = qutip.hinton(dm, color_style="threshold") >>> fig.show() >>> qutip.settings.colorblind_safe = False >>> >>> fig, ax = qutip.hinton(dm, color_style="phase") >>> fig.show()
- matrix_histogram(M, x_basis=None, y_basis=None, limits=None, bar_style='real', color_limits=None, color_style='real', options=None, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Draw a histogram for the matrix M, with the given x and y labels and title.
- Parameters:
- MMatrix of Qobj
The matrix to visualize
- x_basislist of strings, optional
list of x ticklabels
- y_basislist of strings, optional
list of y ticklabels
- limitslist/array with two float numbers, optional
The z-axis limits [min, max]
- bar_stylestring, default=”real”
If set to
"real"
(default), each bar is plotted as the real part of the corresponding matrix elementIf set to
"img"
, each bar is plotted as the imaginary part of the corresponding matrix elementIf set to
"abs"
, each bar is plotted as the absolute value of the corresponding matrix elementIf set to
"phase"
(default), each bar is plotted as the angle of the corresponding matrix element
- color_limitslist/array with two float numbers, optional
The limits of colorbar [min, max]
- color_stylestring, default=”real”
Determines how colors are assigned to each square:
If set to
"real"
(default), each color is chosen according to the real part of the corresponding matrix element.If set to
"img"
, each color is chosen according to the imaginary part of the corresponding matrix element.If set to
"abs"
, each color is chosen according to the absolute value of the corresponding matrix element.If set to
"phase"
, each color is chosen according to the angle of the corresponding matrix element.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
show colorbar
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- optionsdict, optional
A dictionary containing extra options for the plot. The names (keys) and values of the options are described below:
- ‘zticks’list of numbers, optional
A list of z-axis tick locations.
- ‘bars_spacing’float, default=0.1
spacing between bars.
- ‘bars_alpha’float, default=1.
transparency of bars, should be in range 0 - 1
- ‘bars_lw’float, default=0.5
linewidth of bars’ edges.
- ‘bars_edgecolor’color, default=’k’
The colors of the bars’ edges. Examples: ‘k’, (0.1, 0.2, 0.5) or ‘#0f0f0f80’.
- ‘shade’bool, default=True
Whether to shade the dark sides of the bars (True) or not (False). The shading is relative to plot’s source of light.
- ‘azim’float, default=-35
The azimuthal viewing angle.
- ‘elev’float, default=35
The elevation viewing angle.
- ‘stick’bool, default=False
Changes xlim and ylim in such a way that bars next to XZ and YZ planes will stick to those planes. This option has no effect if
ax
is passed as a parameter.- ‘cbar_pad’float, default=0.04
The fraction of the original axes between the colorbar and the new image axes. (i.e. the padding between the 3D figure and the colorbar).
- ‘cbar_to_z’bool, default=False
Whether to set the color of maximum and minimum z-values to the maximum and minimum colors in the colorbar (True) or not (False).
- ‘threshold’: float, optional
Threshold for when bars of smaller height should be transparent. If not set, all bars are colored according to the color map.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- Raises:
- ValueError
Input argument is not valid.
- plot_energy_levels(H_list, h_labels=None, energy_levels=None, N=0, *, fig=None, ax=None)[source]
Plot the energy level diagrams for a list of Hamiltonians. Include up to N energy levels. For each element in H_list, the energy levels diagram for the cummulative Hamiltonian sum(H_list[0:n]) is plotted, where n is the index of an element in H_list.
- Parameters:
- H_listList of Qobj
A list of Hamiltonians.
- h_lablesList of string, optional
A list of xticklabels for each Hamiltonian
- energy_levelsList of string, optional
A list of yticklabels to the left of energy levels of the initial Hamiltonian.
- Nint, default=0
The number of energy levels to plot
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- Returns:
- fig, axtuple
A tuple of the matplotlib figure and axes instances used to produce the figure.
- Raises:
- ValueError
Input argument is not valid.
- plot_expectation_values(results, ylabels=None, *, fig=None, axes=None)[source]
Visualize the results (expectation values) for an evolution solver. results is assumed to be an instance of Result, or a list of Result instances. :param results: List of results objects returned by any of the QuTiP evolution solvers. :type results: (list of)
qutip.solver.Result
:param ylabels: The y-axis labels. List should be of the same length as results. :type ylabels: list of strings, optional :param fig: The Figure canvas in which the plot will be drawn. :type fig: a matplotlib Figure instance, optional :param axes: The axes context in which the plot will be drawn. :type axes: (list of) axes instances, optional- Returns:
- fig, axestuple
A tuple of the matplotlib figure and array of axes instances used to produce the figure.
- plot_fock_distribution(rho, fock_numbers=None, color='green', unit_y_range=True, *, fig=None, ax=None)[source]
Plot the Fock distribution for a density matrix (or ket) that describes an oscillator mode.
- Parameters:
- rhoqutip.Qobj
The density matrix (or ket) of the state to visualize.
- fock_numberslist of strings, optional
list of x ticklabels to represent fock numbers
- colorcolor or list of colors, default=”green”
The colors of the bar faces.
- unit_y_rangebool, default=True
Set y-axis limits [0, 1] or not
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- plot_qubism(ket, theme='light', how='pairs', grid_iteration=1, legend_iteration=0, *, fig=None, ax=None)[source]
Qubism plot for pure states of many qudits. Works best for spin chains, especially with even number of particles of the same dimension. Allows to see entanglement between first 2k particles and the rest.
Note
colorblind_safe does not apply because of its unique colormap
- Parameters:
- ketQobj
Pure state for plotting.
- theme‘light’ or ‘dark’, default=’light’
Set coloring theme for mapping complex values into colors. See: complex_array_to_rgb.
- how‘pairs’, ‘pairs_skewed’ or ‘before_after’, default=’pairs’
Type of Qubism plotting. Options:
‘pairs’ - typical coordinates,
‘pairs_skewed’ - for ferromagnetic/antriferromagnetic plots,
‘before_after’ - related to Schmidt plot (see also: plot_schmidt).
- grid_iterationint, default=1
Helper lines to be drawn on plot. Show tiles for 2*grid_iteration particles vs all others.
- legend_iterationint or ‘grid_iteration’ or ‘all’, default=0
Show labels for first
2*legend_iteration
particles. Option ‘grid_iteration’ sets the same number of particles as for grid_iteration. Option ‘all’ makes label for all particles. Typically it should be 0, 1, 2 or perhaps 3.- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
Notes
See also [1].
References
[1]J. Rodriguez-Laguna, P. Migdal, M. Ibanez Berganza, M. Lewenstein and G. Sierra, Qubism: self-similar visualization of many-body wavefunctions, New J. Phys. 14 053028, arXiv:1112.3560 (2012), open access.
- plot_schmidt(ket, theme='light', splitting=None, labels_iteration=(3, 2), *, fig=None, ax=None)[source]
Plotting scheme related to Schmidt decomposition. Converts a state into a matrix (A_ij -> A_i^j), where rows are first particles and columns - last.
See also: plot_qubism with how=’before_after’ for a similar plot.
Note
colorblind_safe does not apply because of its unique colormap
- Parameters:
- ketQobj
Pure state for plotting.
- theme‘light’ or ‘dark’, default=’light’
Set coloring theme for mapping complex values into colors. See: complex_array_to_rgb.
- splittingint, optional
Plot for a number of first particles versus the rest. If not given, it is (number of particles + 1) // 2.
- labels_iterationint or pair of ints, default=(3,2)
Number of particles to be shown as tick labels, for first (vertical) and last (horizontal) particles, respectively.
- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- plot_spin_distribution(P, THETA, PHI, projection='2d', *, cmap=None, colorbar=False, fig=None, ax=None)[source]
Plots a spin distribution (given as meshgrid data).
- Parameters:
- Pmatrix
Distribution values as a meshgrid matrix.
- THETAmatrix
Meshgrid matrix for the theta coordinate. Its range is between 0 and pi
- PHImatrix
Meshgrid matrix for the phi coordinate. Its range is between 0 and 2*pi
- projection: string {‘2d’, ‘3d’}, default=’2d’
Specify whether the spin distribution function is to be plotted as a 2D projection where the surface of the unit sphere is mapped on the unit disk (‘2d’) or surface plot (‘3d’).
- cmapa matplotlib cmap instance, optional
The colormap.
- colorbarbool, default=False
Whether (True) or not (False) a colorbar should be attached to the Wigner function graph.
- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- plot_wigner(rho, xvec=None, yvec=None, method='clenshaw', projection='2d', *, cmap=None, colorbar=False, fig=None, ax=None)[source]
Plot the the Wigner function for a density matrix (or ket) that describes an oscillator mode.
- Parameters:
- rhoqutip.Qobj
The density matrix (or ket) of the state to visualize.
- xvecarray_like, optional
x-coordinates at which to calculate the Wigner function.
- yvecarray_like, optional
y-coordinates at which to calculate the Wigner function. Does not apply to the ‘fft’ method.
- methodstring {‘clenshaw’, ‘iterative’, ‘laguerre’, ‘fft’},
default=’clenshaw’ The method used for calculating the wigner function. See the documentation for qutip.wigner for details.
- projection: string {‘2d’, ‘3d’}, default=’2d’
Specify whether the Wigner function is to be plotted as a contour graph (‘2d’) or surface plot (‘3d’).
- cmapa matplotlib cmap instance, optional
The colormap.
- colorbarbool, default=False
Whether (True) or not (False) a colorbar should be attached to the Wigner function graph.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
- plot_wigner_sphere(wigner, reflections=False, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Plots a coloured Bloch sphere.
- Parameters:
- wignera wigner transformation
The wigner transformation at steps different theta and phi.
- reflectionsbool, default=False
If the reflections of the sphere should be plotted as well.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The ax context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
Notes
Special thanks to Russell P Rundle for writing this function.
- sphereplot(values, theta, phi, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Plots a matrix of values on a sphere
- Parameters:
- valuesarray
Data set to be plotted
- thetafloat
Angle with respect to z-axis. Its range is between 0 and pi
- phifloat
Angle in x-y plane. Its range is between 0 and 2*pi
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- Returns:
- fig, outputtuple
A tuple of the matplotlib figure and the axes instance or animation instance used to produce the figure.
Functions to animate results of quantum dynamics simulations,
- anim_fock_distribution(rhos, fock_numbers=None, color='green', unit_y_range=True, *, fig=None, ax=None)[source]
Animation of the Fock distribution for a density matrix (or ket) that describes an oscillator mode.
- Parameters:
- rhos
qutip.solver.Result
or list ofqutip.Qobj
The density matrix (or ket) of the state to visualize.
- fock_numberslist of strings, optional
list of x ticklabels to represent fock numbers
- colorcolor or list of colors, default=”green”
The colors of the bar faces.
- unit_y_rangebool, default=True
Set y-axis limits [0, 1] or not
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- rhos
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- anim_hinton(rhos, x_basis=None, y_basis=None, color_style='scaled', label_top=True, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Draws an animation of Hinton diagram.
- Parameters:
- rhos
qutip.solver.Result
or list ofqutip.Qobj
Input density matrix or superoperator.
Note
Hinton plots of superoperators are currently only supported for qubits.
- x_basislist of strings, optional
list of x ticklabels to represent x basis of the input.
- y_basislist of strings, optional
list of y ticklabels to represent y basis of the input.
- color_stylestring, default=”scaled”
Determines how colors are assigned to each square:
If set to
"scaled"
(default), each color is chosen by passing the absolute value of the corresponding matrix element into cmap with the sign of the real part.If set to
"threshold"
, each square is plotted as the maximum of cmap for the positive real part and as the minimum for the negative part of the matrix element; note that this generalizes “threshold” to complex numbers.If set to
"phase"
, each color is chosen according to the angle of the corresponding matrix element.
- label_topbool, default=True
If True, x ticklabels will be placed on top, otherwise they will appear below the plot.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The ax context in which the plot will be drawn.
- rhos
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- Raises:
- ValueError
Input argument is not a quantum object.
- anim_matrix_histogram(Ms, x_basis=None, y_basis=None, limits=None, bar_style='real', color_limits=None, color_style='real', options=None, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Draw an animation of a histogram for the matrix M, with the given x and y labels.
- Parameters:
- Mslist of matrices or
qutip.solver.Result
The matrix to visualize
- x_basislist of strings, optional
list of x ticklabels
- y_basislist of strings, optional
list of y ticklabels
- limitslist/array with two float numbers, optional
The z-axis limits [min, max]
- bar_stylestring, default=”real”
If set to
"real"
(default), each bar is plotted as the real part of the corresponding matrix elementIf set to
"img"
, each bar is plotted as the imaginary part of the corresponding matrix elementIf set to
"abs"
, each bar is plotted as the absolute value of the corresponding matrix elementIf set to
"phase"
(default), each bar is plotted as the angle of the corresponding matrix element
- color_limitslist/array with two float numbers, optional
The limits of colorbar [min, max]
- color_stylestring, default=”real”
Determines how colors are assigned to each square:
If set to
"real"
(default), each color is chosen according to the real part of the corresponding matrix element.If set to
"img"
, each color is chosen according to the imaginary part of the corresponding matrix element.If set to
"abs"
, each color is chosen according to the absolute value of the corresponding matrix element.If set to
"phase"
, each color is chosen according to the angle of the corresponding matrix element.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
show colorbar
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- optionsdict, optional
A dictionary containing extra options for the plot. The names (keys) and values of the options are described below:
- ‘zticks’list of numbers, optional
A list of z-axis tick locations.
- ‘bars_spacing’float, default=0.1
spacing between bars.
- ‘bars_alpha’float, default=1.
transparency of bars, should be in range 0 - 1
- ‘bars_lw’float, default=0.5
linewidth of bars’ edges.
- ‘bars_edgecolor’color, default=’k’
The colors of the bars’ edges. Examples: ‘k’, (0.1, 0.2, 0.5) or ‘#0f0f0f80’.
- ‘shade’bool, default=True
Whether to shade the dark sides of the bars (True) or not (False). The shading is relative to plot’s source of light.
- ‘azim’float, default=-35
The azimuthal viewing angle.
- ‘elev’float, default=35
The elevation viewing angle.
- ‘stick’bool, default=False
Changes xlim and ylim in such a way that bars next to XZ and YZ planes will stick to those planes. This option has no effect if
ax
is passed as a parameter.- ‘cbar_pad’float, default=0.04
The fraction of the original axes between the colorbar and the new image axes. (i.e. the padding between the 3D figure and the colorbar).
- ‘cbar_to_z’bool, default=False
Whether to set the color of maximum and minimum z-values to the maximum and minimum colors in the colorbar (True) or not (False).
- ‘threshold’: float, optional
Threshold for when bars of smaller height should be transparent. If not set, all bars are colored according to the color map.
- Mslist of matrices or
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- Raises:
- ValueError
Input argument is not valid.
- anim_qubism(kets, theme='light', how='pairs', grid_iteration=1, legend_iteration=0, *, fig=None, ax=None)[source]
Animation of Qubism plot for pure states of many qudits. Works best for spin chains, especially with even number of particles of the same dimension. Allows to see entanglement between first 2k particles and the rest.
Note
colorblind_safe does not apply because of its unique colormap
- Parameters:
- kets
qutip.solver.Result
or list ofqutip.Qobj
Pure states for animation.
- theme‘light’ or ‘dark’, default=’light’
Set coloring theme for mapping complex values into colors. See: complex_array_to_rgb.
- how‘pairs’, ‘pairs_skewed’ or ‘before_after’, default=’pairs’
Type of Qubism plotting. Options:
‘pairs’ - typical coordinates,
‘pairs_skewed’ - for ferromagnetic/antriferromagnetic plots,
‘before_after’ - related to Schmidt plot (see also: plot_schmidt).
- grid_iterationint, default=1
Helper lines to be drawn on plot. Show tiles for 2*grid_iteration particles vs all others.
- legend_iterationint or ‘grid_iteration’ or ‘all’, default=0
Show labels for first
2*legend_iteration
particles. Option ‘grid_iteration’ sets the same number of particles as for grid_iteration. Option ‘all’ makes label for all particles. Typically it should be 0, 1, 2 or perhaps 3.- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- kets
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
Notes
See also [1].
References
[1]J. Rodriguez-Laguna, P. Migdal, M. Ibanez Berganza, M. Lewenstein and G. Sierra, Qubism: self-similar visualization of many-body wavefunctions, New J. Phys. 14 053028, arXiv:1112.3560 (2012), open access.
- anim_schmidt(kets, theme='light', splitting=None, labels_iteration=(3, 2), *, fig=None, ax=None)[source]
Animation of Schmidt decomposition. Converts a state into a matrix (A_ij -> A_i^j), where rows are first particles and columns - last.
See also: plot_qubism with how=’before_after’ for a similar plot.
Note
colorblind_safe does not apply because of its unique colormap
- Parameters:
- ket
qutip.solver.Result
or list ofqutip.Qobj
Pure states for animation.
- theme‘light’ or ‘dark’, default=’light’
Set coloring theme for mapping complex values into colors. See: complex_array_to_rgb.
- splittingint, optional
Plot for a number of first particles versus the rest. If not given, it is (number of particles + 1) // 2.
- labels_iterationint or pair of ints, default=(3,2)
Number of particles to be shown as tick labels, for first (vertical) and last (horizontal) particles, respectively.
- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- ket
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- anim_sphereplot(V, theta, phi, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
animation of a matrices of values on a sphere
- Parameters:
- Vlist of array instances
Data set to be plotted
- thetafloat
Angle with respect to z-axis. Its range is between 0 and pi
- phifloat
Angle in x-y plane. Its range is between 0 and 2*pi
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- anim_spin_distribution(Ps, THETA, PHI, projection='2d', *, cmap=None, colorbar=False, fig=None, ax=None)[source]
Animation of a spin distribution (given as meshgrid data).
- Parameters:
- Pslist of matrices
Distribution values as a meshgrid matrix.
- THETAmatrix
Meshgrid matrix for the theta coordinate. Its range is between 0 and pi
- PHImatrix
Meshgrid matrix for the phi coordinate. Its range is between 0 and 2*pi
- projection: string {‘2d’, ‘3d’}, default=’2d’
Specify whether the spin distribution function is to be plotted as a 2D projection where the surface of the unit sphere is mapped on the unit disk (‘2d’) or surface plot (‘3d’).
- cmapa matplotlib cmap instance, optional
The colormap.
- colorbarbool, default=False
Whether (True) or not (False) a colorbar should be attached to the Wigner function graph.
- figa matplotlib figure instance, optional
The figure canvas on which the plot will be drawn.
- axa matplotlib axis instance, optional
The axis context in which the plot will be drawn.
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- anim_wigner(rhos, xvec=None, yvec=None, method='clenshaw', projection='2d', *, cmap=None, colorbar=False, fig=None, ax=None)[source]
Animation of the Wigner function for a density matrix (or ket) that describes an oscillator mode.
- Parameters:
- rhos
qutip.solver.Result
or list ofqutip.Qobj
The density matrix (or ket) of the state to visualize.
- xvecarray_like, optional
x-coordinates at which to calculate the Wigner function.
- yvecarray_like, optional
y-coordinates at which to calculate the Wigner function. Does not apply to the ‘fft’ method.
- methodstring {‘clenshaw’, ‘iterative’, ‘laguerre’, ‘fft’},
default=’clenshaw’ The method used for calculating the wigner function. See the documentation for qutip.wigner for details.
- projection: string {‘2d’, ‘3d’}, default=’2d’
Specify whether the Wigner function is to be plotted as a contour graph (‘2d’) or surface plot (‘3d’).
- cmapa matplotlib cmap instance, optional
The colormap.
- colorbarbool, default=False
Whether (True) or not (False) a colorbar should be attached to the Wigner function graph.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The axes context in which the plot will be drawn.
- rhos
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
- anim_wigner_sphere(wigners, reflections=False, *, cmap=None, colorbar=True, fig=None, ax=None)[source]
Animate a coloured Bloch sphere.
- Parameters:
- wignerslist of transformations
The wigner transformation at steps different theta and phi.
- reflectionsbool, default=False
If the reflections of the sphere should be plotted as well.
- cmapa matplotlib colormap instance, optional
Color map to use when plotting.
- colorbarbool, default=True
Whether (True) or not (False) a colorbar should be attached.
- figa matplotlib Figure instance, optional
The Figure canvas in which the plot will be drawn.
- axa matplotlib axes instance, optional
The ax context in which the plot will be drawn.
- Returns:
- fig, anituple
A tuple of the matplotlib figure and the animation instance used to produce the figure.
Notes
Special thanks to Russell P Rundle for writing this function.
This module contains utility functions that enhance Matplotlib in one way or another.
- complex_phase_cmap()[source]
Create a cyclic colormap for representing the phase of complex variables
- Returns:
- cmap
A matplotlib linear segmented colormap.
- wigner_cmap(W, levels=1024, shift=0, max_color='#09224F', mid_color='#FFFFFF', min_color='#530017', neg_color='#FF97D4', invert=False)[source]
A custom colormap that emphasizes negative values by creating a nonlinear colormap.
- Parameters:
- Warray
Wigner function array, or any array.
- levelsint
Number of color levels to create.
- shiftfloat
Shifts the value at which Wigner elements are emphasized. This parameter should typically be negative and small (i.e -1e-5).
- max_colorstr
String for color corresponding to maximum value of data. Accepts any string format compatible with the Matplotlib.colors.ColorConverter.
- mid_colorstr
Color corresponding to zero values. Accepts any string format compatible with the Matplotlib.colors.ColorConverter.
- min_colorstr
Color corresponding to minimum data values. Accepts any string format compatible with the Matplotlib.colors.ColorConverter.
- neg_colorstr
Color that starts highlighting negative values. Accepts any string format compatible with the Matplotlib.colors.ColorConverter.
- invertbool
Invert the color scheme for negative values so that smaller negative values have darker color.
- Returns:
- Returns a Matplotlib colormap instance for use in plotting.
Notes
The ‘shift’ parameter allows you to vary where the colormap begins to highlight negative colors. This is beneficial in cases where there are small negative Wigner elements due to numerical round-off and/or truncation.
Quantum Process Tomography
- qpt(U, op_basis_list)[source]
Calculate the quantum process tomography chi matrix for a given (possibly nonunitary) transformation matrix U, which transforms a density matrix in vector form according to:
vec(rho) = U * vec(rho0)
or
rho = unstack_columns(U * stack_columns(rho0))
U can be calculated for an open quantum system using the QuTiP propagator function.
- Parameters:
- UQobj
Transformation operator. Can be calculated using QuTiP propagator function.
- op_basis_listlist
A list of Qobj’s representing the basis states.
- Returns:
- chiarray
QPT chi matrix
- qpt_plot(chi, lbls_list, title=None, fig=None, axes=None)[source]
Visualize the quantum process tomography chi matrix. Plot the real and imaginary parts separately.
- Parameters:
- chiarray
Input QPT chi matrix.
- lbls_listlist
List of labels for QPT plot axes.
- titlestring
Plot title.
- figfigure instance
User defined figure instance used for generating QPT plot.
- axeslist of figure axis instance
User defined figure axis instance (list of two axes) used for generating QPT plot.
- Returns:
- fig, axtuple
A tuple of the matplotlib figure and axes instances used to produce the figure.
- qpt_plot_combined(chi, lbls_list, title=None, fig=None, ax=None, figsize=(8, 6), threshold=None)[source]
Visualize the quantum process tomography chi matrix. Plot bars with height and color corresponding to the absolute value and phase, respectively.
- Parameters:
- chiarray
Input QPT chi matrix.
- lbls_listlist
List of labels for QPT plot axes.
- titlestring
Plot title.
- figfigure instance
User defined figure instance used for generating QPT plot.
- axfigure axis instance
User defined figure axis instance used for generating QPT plot (alternative to the fig argument).
- threshold: float (None)
Threshold for when bars of smaller height should be transparent. If not set, all bars are colored according to the color map.
- Returns:
- fig, axtuple
A tuple of the matplotlib figure and axes instances used to produce the figure.
Non-Markovian Solvers
This module contains an implementation of the non-Markovian transfer tensor method (TTM), introduced in [1].
[1] Javier Cerrillo and Jianshu Cao, Phys. Rev. Lett 112, 110401 (2014)
- ttmsolve(dynmaps, state0, times, e_ops=[], num_learning=0, options=None)[source]
Expand time-evolution using the Transfer Tensor Method [1], based on a set of precomputed dynamical maps.
- Parameters:
- dynmapslist of
qutip.Qobj
, callable List of precomputed dynamical maps (superoperators) for the first times of
times
or a callback function that returns the superoperator at a given time.- state0
qutip.Qobj
Initial density matrix or state vector (ket).
- timesarray_like
List of times \(t_n\) at which to compute results. Must be uniformily spaced.
- e_ops
qutip.qobj
, callable, or list. Single operator or list of operators for which to evaluate expectation values or callable or list of callable. Callable signature must be, f(t: float, state: Qobj). See
expect
for more detail of operator expectation.- num_learningint
Number of times used to construct the dynmaps operators when
dynmaps
is a callable.- optionsdictionary
Dictionary of options for the solver.
store_final_state : bool Whether or not to store the final state of the evolution in the result class.
store_states : bool, None Whether or not to store the state vectors or density matrices. On None the states will be saved if no expectation operators are given.
normalize_output : bool Normalize output state to hide ODE numerical errors.
threshold : float Threshold for halting. Halts if \(||T_{n}-T_{n-1}||\) is below treshold.
- dynmapslist of
- Returns:
- output:
qutip.solver.Result
An instance of the class
qutip.solver.Result
.
[1]Javier Cerrillo and Jianshu Cao, Phys. Rev. Lett 112, 110401 (2014) ..
- output:
Utility Functions
Utility Functions
This module contains utility functions that are commonly needed in other qutip modules.
- clebsch(j1, j2, j3, m1, m2, m3)[source]
Calculates the Clebsch-Gordon coefficient for coupling (j1,m1) and (j2,m2) to give (j3,m3).
- Parameters:
- j1float
Total angular momentum 1.
- j2float
Total angular momentum 2.
- j3float
Total angular momentum 3.
- m1float
z-component of angular momentum 1.
- m2float
z-component of angular momentum 2.
- m3float
z-component of angular momentum 3.
- Returns:
- cg_coefffloat
Requested Clebsch-Gordan coefficient.
- convert_unit(value, orig='meV', to='GHz')[source]
Convert an energy from unit orig to unit to.
- Parameters:
- valuefloat / array
The energy in the old unit.
- origstring
The name of the original unit (“J”, “eV”, “meV”, “GHz”, “mK”)
- tostring
The name of the new unit (“J”, “eV”, “meV”, “GHz”, “mK”)
- Returns:
- value_new_unitfloat / array
The energy in the new unit.
- n_thermal(w, w_th)[source]
Return the number of photons in thermal equilibrium for an harmonic oscillator mode with frequency ‘w’, at the temperature described by ‘w_th’ where \(\omega_{\rm th} = k_BT/\hbar\).
- Parameters:
- wfloat or array
Frequency of the oscillator.
- w_thfloat
The temperature in units of frequency (or the same units as w).
- Returns:
- n_avgfloat or array
Return the number of average photons in thermal equilibrium for a an oscillator with the given frequency and temperature.
File I/O Functions
- file_data_read(filename, sep=None)[source]
Retrieves an array of data from the requested file.
- Parameters:
- filenamestr or pathlib.Path
Name of file containing reqested data.
- sepstr
Seperator used to store data.
- Returns:
- dataarray_like
Data from selected file.
- file_data_store(filename, data, numtype='complex', numformat='decimal', sep=',')[source]
Stores a matrix of data to a file to be read by an external program.
- Parameters:
- filenamestr or pathlib.Path
Name of data file to be stored, including extension.
- data: array_like
Data to be written to file.
- numtypestr {‘complex, ‘real’}
Type of numerical data.
- numformatstr {‘decimal’,’exp’}
Format for written data.
- sepstr
Single-character field seperator. Usually a tab, space, comma, or semicolon.
Parallelization
This module provides functions for parallel execution of loops and function mappings, using the builtin Python module multiprocessing or the loky parallel execution library.
- parallel_map(task, values, task_args=None, task_kwargs=None, reduce_func=None, map_kw=None, progress_bar=None, progress_bar_kwargs={})[source]
Parallel execution of a mapping of values to the function task. This is functionally equivalent to:
result = [task(value, *task_args, **task_kwargs) for value in values]
- Parameters:
- taska Python function
The function that is to be called for each value in
task_vec
.- valuesarray / list
The list or array of values for which the
task
function is to be evaluated.- task_argslist / dictionary
The optional additional argument to the
task
function.- task_kwargslist / dictionary
The optional additional keyword argument to the
task
function.- reduce_funcfunc (optional)
If provided, it will be called with the output of each tasks instead of storing a them in a list. Note that the order in which results are passed to
reduce_func
is not defined. It should return None or a number. When returning a number, it represent the estimation of the number of task left. On a return <= 0, the map will end early.- progress_barstring
Progress bar options’s string for showing progress.
- progress_bar_kwargsdict
Options for the progress bar.
- map_kw: dict (optional)
Dictionary containing entry for: - timeout: float, Maximum time (sec) for the whole map. - job_timeout: float, Maximum time (sec) for each job in the map. - num_cpus: int, Number of job to run at once. - fail_fast: bool, Raise an error at the first.
- Returns:
- resultlist
The result list contains the value of
task(value, *task_args, **task_kwargs)
for each value invalues
. If areduce_func
is provided, and empty list will be returned.
- serial_map(task, values, task_args=None, task_kwargs=None, reduce_func=None, map_kw=None, progress_bar=None, progress_bar_kwargs={})[source]
Serial mapping function with the same call signature as parallel_map, for easy switching between serial and parallel execution. This is functionally equivalent to:
result = [task(value, *task_args, **task_kwargs) for value in values]
This function work as a drop-in replacement of
qutip.parallel_map
.- Parameters:
- taska Python function
The function that is to be called for each value in
task_vec
.- valuesarray / list
The list or array of values for which the
task
function is to be evaluated.- task_argslist / dictionary
The optional additional argument to the
task
function.- task_kwargslist / dictionary
The optional additional keyword argument to the
task
function.- reduce_funcfunc (optional)
If provided, it will be called with the output of each tasks instead of storing a them in a list. It should return None or a number. When returning a number, it represent the estimation of the number of task left. On a return <= 0, the map will end early.
- progress_barstring
Progress bar options’s string for showing progress.
- progress_bar_kwargsdict
Options for the progress bar.
- map_kw: dict (optional)
Dictionary containing: - timeout: float, Maximum time (sec) for the whole map. - fail_fast: bool, Raise an error at the first.
- Returns:
- resultlist
The result list contains the value of
task(value, *task_args, **task_kwargs)
for each value invalues
. If areduce_func
is provided, and empty list will be returned.
Semidefinite Programming
IPython Notebook Tools
This module contains utility functions for using QuTiP with IPython notebooks.
- parallel_map(task, values, task_args=None, task_kwargs=None, client=None, view=None, progress_bar=None, show_scheduling=False, **kwargs)[source]
Call the function
task
for each value invalues
using a cluster of IPython engines. The functiontask
should have the signaturetask(value, *args, **kwargs)
.The
client
andview
are the IPython.parallel client and load-balanced view that will be used in the parfor execution. If these areNone
, new instances will be created.- Parameters:
- task: a Python function
The function that is to be called for each value in
task_vec
.- values: array / list
The list or array of values for which the
task
function is to be evaluated.- task_args: list / dictionary
The optional additional argument to the
task
function.- task_kwargs: list / dictionary
The optional additional keyword argument to the
task
function.- client: IPython.parallel.Client
The IPython.parallel Client instance that will be used in the parfor execution.
- view: a IPython.parallel.Client view
The view that is to be used in scheduling the tasks on the IPython cluster. Preferably a load-balanced view, which is obtained from the IPython.parallel.Client instance client by calling, view = client.load_balanced_view().
- show_scheduling: bool {False, True}, default False
Display a graph showing how the tasks (the evaluation of
task
for for the value intask_vec1
) was scheduled on the IPython engine cluster.- show_progressbar: bool {False, True}, default False
Display a HTML-based progress bar during the execution of the parfor loop.
- Returns:
- resultlist
The result list contains the value of
task(value, task_args, task_kwargs)
for each value invalues
.
- version_table(verbose=False)[source]
Print an HTML-formatted table with version numbers for QuTiP and its dependencies. Use it in a IPython notebook to show which versions of different packages that were used to run the notebook. This should make it possible to reproduce the environment and the calculation later on.
- Returns:
- version_table: string
Return an HTML-formatted string containing version information for QuTiP dependencies.
Miscellaneous
- about()[source]
About box for QuTiP. Gives version numbers for QuTiP, NumPy, SciPy, Cython, and MatPlotLib.
- simdiag(ops, evals: bool = True, *, tol: float = 1e-14, safe_mode: bool = True)[source]
Simultaneous diagonalization of commuting Hermitian matrices.
- Parameters:
- opslist/array
list
orarray
of qobjs representing commuting Hermitian operators.- evalsbool [True]
Whether to return the eigenvalues for each ops and eigenvectors or just the eigenvectors.
- tolfloat [1e-14]
Tolerance for detecting degenerate eigenstates.
- safe_modebool [True]
Whether to check that all ops are Hermitian and commuting. If set to
False
and operators are not commuting, the eigenvectors returned will often be eigenvectors of only the first operator.
- Returns:
- eigstuple
Tuple of arrays representing eigvecs and eigvals of quantum objects corresponding to simultaneous eigenvectors and eigenvalues for each operator.