如何使用mplot3d和numpy数组绘制3D网格？

Question

我正在尝试绘制具有整数值的3D NumPy数组。

import numpy as np

def build(self):
    grid = np.empty((10,10,10))
    grid = grid.astype(np.int)
    grid.fill(-1)
    return grid

先前的方法将构建3D网格并以-1值填充。 在这种情况下，-1表示空白单元格。

在网格中放置3个元素，例如，在（0,0,0），（0,0,1）y（5,5,5）的位置上，绘制后我看不到相同数量的元素。 我在这里附上图片 。

图像的右边有第四个元素（具有较小的色调）。

我绘制网格的代码如下所示：

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

def plot_grid(self, np_grid):
        fig = plt.figure()
        ax = fig.add_subplot(111, projection='3d')
        x = np_grid.sum(0)
        y = np_grid.sum(1)
        z = np_grid.sum(2)
        ax.scatter(x, y, -z, zdir='z', c='red')
        plt.show()

我认为我没有正确获得x,y,z向量。 我怎样才能正确地从ndarray获得x,y,z ？

Answer 1

我相信您的问题是您需要将x，y和z作为它们自己的数组传递：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter((0,0,5),(0,0,5),(0,1,5),zdir='z', c='red')

这应该给你这样的情节：