在python

Question

我想在3D图中表示二进制3D矩阵（如果可能，不能使用mayavi.mlab）。在矩阵具有1 a点的每个位置（x，y，z）应该绘制。 我的矩阵按以下方式构建：

import numpy as np
size = 21
my_matrix = np.zeros(shape = (size, size, size))
random_location_1 = (1,1,2)
random_location_2 = (3,5,8)
my_matrix[random_location_1] = 1
my_matrix[random_location_2] = 1

现在在坐标（1,1,2）和（3,5,8）处应该可以看到一个点，其他地方只是空白区域。 有没有办法做到这一点（例如使用matplotlib？）

Answer 1

听起来你需要散点图。看看this mplot3d教程。对我来说这很有效：

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


fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

size = 21
m = np.zeros(shape = (size, size, size))
random_location_1 = (1,1,2)
random_location_2 = (3,5,8)
m[random_location_1] = 1
m[random_location_2] = 1

pos = np.where(m==1)
ax.scatter(pos[0], pos[1], pos[2], c='black')
plt.show()