难以在 matplotlib 中仅制作一个 Line3D 对象

Question

我正在制作类似于 here 的 3D 线对象动画，这实际上来自 here。

我只想制作一行的动画，而不是很多行，而且我不知道这里发生了什么。我认为这与sum() 的工作方式有关。

我怎样才能只做一个，为什么会失败？

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

fig = plt.figure()
ax = fig.add_axes([0, 0, 1, 1], projection='3d')
colors = plt.cm.jet([0.3, 0.7])

LINES = sum([ax.plot([],[],[], '-', c=c) for c in colors], [])
line = LINES[0]
line.set_data([], [])                 # THIS WORKS

LINES = [ax.plot([],[],[], '-', c=colors[0])]
line = LINES[0]
line.set_data([], [])                 # THIS FAILS

所以我可以进一步减少我的问题：

a = [1]
print "type(a[0]): ", type(a[0])   # expect <type 'int'>

b = [ax.plot([],[],[], '-', c=colors[0])]
print "type(b[0]): ", type(b[0])   # expect <class 'mpl_toolkits.mplot3d.art3d.Line3D'>

给予

<type 'int'>
<type 'list'>

Answer 1

这里的关键点是 ax.plot 返回 list 的 Line3D 对象（即使在您的示例中它是一个包含单个成员的列表），而不是 Line3D object itself。

因此，在您的第一个示例中，LINES 是 Line3D 对象的 list，在第二个示例中，它是 lists 的 list。因此，在失败的代码中，line 是一个包含单个 Line3D 对象的 list。您尝试在该list 对象上调用set_data，python 告诉您list 没有该功能。你可以通过line = LINES[0][0] 来实现它，或者干脆做

LINES = ax.plot([],[],[], '-', c=colors[0])[0]

---

要获取您的 example code here 并使其仅绘制一行，请更改行：

# set up lines and points
lines = sum([ax.plot([], [], [], '-', c=c)
             for c in colors], [])
pts = sum([ax.plot([], [], [], 'o', c=c)
           for c in colors], [])

阅读

# set up lines and points
lines = ax.plot([], [], [], '-', c=colors[0])
pts = ax.plot([], [], [], 'o', c=colors[0])

sum 只是一种将推导式的所有成员附加到一个大列表中的方法。上述更改从颜色图中的每种颜色一行变为单行。