首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >如何在matplotlib上为3D绘图绘制单独的线条

如何在matplotlib上为3D绘图绘制单独的线条
EN

Stack Overflow用户
提问于 2021-07-30 15:03:10
回答 1查看 31关注 0票数 0

我有三个变量:

代码语言:javascript
复制
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线。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-07-30 15:14:44

如果你使用numpy数组,你可以使用np.where来获得每个唯一的X值出现的索引。然后使用这些索引对每个ax.plot3D调用的xsyszs进行切片:

代码语言:javascript
复制
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()

输出:

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68593170

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档