使用Python Matploltlib绘制3D点

时间:2017-01-30 23:40:34

标签: python matplotlib 3d

我想创建一个小模拟,我想我知道如何,但为了实际看到发生了什么,我需要将其可视化。

我从一个5x5x5阵列开始,我想填充值。

data = numpy.zeros(shape=(5,5,5))
data[:,:,0]=4
data[:,:,1]=3
data[:,:,2]=2
data[:,:,3]=1
data[:,:,4]=0

这应创建一个在向上方向上具有渐变的立方体(如果第三个轴是z)。

现在,我该如何策划这个?我不想要表面图或线框。只需在每个坐标上指向点,也可以按值进行颜色编码或跨越。

作为测试,我尝试使用

绘制所有坐标
ax.scatter(numpy.arange(5),numpy.arange(5),numpy.arange(5))

但这只会绘制一个由5个点组成的直线。

那么......我怎样才能获得我想创造的125个点?

THX。

1 个答案:

答案 0 :(得分:0)

您可以使用以下颜色对值进行编码:

x = np.arange(5)
X, Y, Z = np.meshgrid(x,x,x)

v = np.arange(125)
ax.scatter(X,Y,Z, c=v)

有关文档,请参阅here