如何在标准图中直接使用matplotlib中的Axes3D来避免flake8错误

Question

使用典型的3D图时,如下所示:

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.gca(projection='3d')

flake8 报告预期的错误:

./tools.py:62:9: F401 'mpl_toolkits.mplot3d.Axes3D' imported but unused

我知道使用# NOQA评论可以避免这种情况.但是有没有不同的方法来配置图中的投影,以便使用Axes3D对象？

Answer 1

如果这只是实际使用导入至少一次,你可以这样做

ax = fig.gca(projection=Axes3D.name)

因为"3d"是名Axes3D通过其注册到投影列表类.