如何使用QuTiP绘制Bloch球面时给出aplot标题

Question

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

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的调用中设置了ax和Bloch，它应该已经自动绘制到正确的轴上，但是render功能已经设置默认为fig=None, axis=None。这可能是一个小错误。