在matplotlib 3D中标记第一个和最后一个数据点

Question

我在3D图中绘制了一组数据点，我想用不同的颜色标记第一个和最后一个数据点，并用图例标记它们。我该怎么做？

我使用的代码是

from mpl_toolkits.mplot3d import Axes3D

x = np.array([0,1,2,3])
y = np.array([0,1,2,3])
z = np.array([0,1,2,3])

fig = plt.figure()
ax = fig.add_subplot(111,projection='3d')
ax.plot(x,y,z,'o-',markersize=5)
plt.show()

Answer 1

您可以重绘图上的第一个和最后一个点，并在给它们颜色时标记它们。

from mpl_toolkits.mplot3d import Axes3D

x = np.array([0,1,2,3])
y = np.array([0,1,2,3])
z = np.array([0,1,2,3])
fig = plt.figure()
ax = fig.add_subplot(111,projection='3d')
ax.plot(x[:1], y[:1], z[:1], 'o-',c='green', label="first", zorder=2)
ax.plot(x[-1:], y[-1:], z[-1:], 'o-',c='coral', label="last", zorder=2)
ax.plot(x,y,z,'o-',markersize=5, zorder=1)
ax.legend()
plt.show()

输出：