python - 如何在使用 QuTiP 绘制 Bloch 球体时给出 aplot 标题

Question

我正在使用 QuTiP 绘制一个 Bloch 球体图。我想给它一个标题。我怎样才能做到这一点？我在谷歌上搜索但找不到答案。

Answer 1

您需要在 matploltib 中将 Bloch 球体渲染到 3D 图形中的轴上。以下是如何执行此操作的示例：

import matplotlib.pyplot as plt
import quitp

# needs Axes3D object to activate the '3d' projection
from mpl_toolkits.mplot3d import Axes3D

fig, ax = plt.subplots(figsize=(5, 5), subplot_kw=dict(projection='3d'))

ax.axis('square') # to get a nice circular plot

b1 = qutip.Bloch(fig=fig, axes=ax)
b1.add_states(qutip.sigmax()/2)
b1.zlabel = ['z', '']
b1.render(fig=fig, axes=ax) # render to the correct subplot 

# set title for the axis
ax.set_title('TITLE goes here', y=1.1, fontsize=20)

# You can anything else you want to the axis as well!
ax.annotate('TEXT', xy=(0.1, 0.9), xytext=(0.1, 0.7), xycoords='axes fraction',
            fontsize=15, color='r', ha='center',)

plt.show()

这是输出：

理想情况下，一旦我们在对 的调用中设置fig和，它应该会自动绘制到正确的轴上，但该函数默认设置为。这可能是一个小错误。axBlochrenderfig=None, axis=None