我有三个变量:
xs = [1,1,1,2,2,2,3,3,3],
ys = [1,2,3,1,2,3,1,2,3],
zs = [1,4,9,2,8,18,4,16,36]我想通过固定xs来绘制一个有3条单独的线的图,例如,对于x=2,我想看到y和zs (其中x= 2)的线,在这种情况下应该是ys = [1,2,3], zs = [2,8,18]。我希望能够看到x中每个唯一x的不同线,但在相同的图中。有人知道我是怎么做的吗?
我尝试过使用ax.plot3D(xs,ys,zs),但正如你所期望的那样,它连接了多个点,所以它是一条连续的3D线。
发布于 2021-07-30 15:14:44
如果你使用numpy数组,你可以使用np.where来获得每个唯一的X值出现的索引。然后使用这些索引对每个ax.plot3D调用的xs、ys和zs进行切片:
import matplotlib.pyplot as plt
import numpy as np
# Create numpy arrays
xs = np.array([1,1,1,2,2,2,3,3,3])
ys = np.array([1,2,3,1,2,3,1,2,3])
zs = np.array([1,4,9,2,8,18,4,16,36])
# Create Figure and 3D Axes
fig = plt.figure()
ax = fig.add_subplot(projection='3d')
# Iterate over unique values in xs array
for unique_x in np.unique(xs):
# Slice arrays and plot
xvals = xs[np.where(xs == unique_x)]
yvals = ys[np.where(xs == unique_x)]
zvals = zs[np.where(xs == unique_x)]
ax.plot3D(xvals, yvals, zvals)
# Display Figure
plt.show()输出:

https://stackoverflow.com/questions/68593170
复制相似问题