三维数组中每个元素的索引值

Question

我使用numpy mgrid命令创建了一个3D数组，因此每个元素都有一个特定的值，索引保留空间信息。例如，如果在z轴（三维）上求和，则生成的2D数组可以在matplotlib中使用imshow（）函数，以获得具有不同二进制像素值的图像。在

我的问题是：如何获得网格中每个元素的索引值（a、b、c）？在

我需要使用索引值来计算每个点到网格原点的相对角度。（例如θ=sin-1（sqrt（x^2+y^2）/sqrt（x^2+y^2+z^2））

也许这可以转换成另一个三维网格，其中每个元素都是数组[a，b，c]？在

Answer 1

我不太清楚您的意思，但是如果您要寻找包含索引x、y和z的3d数组，那么下面的内容可能适合您的需要；假设您的数据保存在一个名为“abc”的3d数组中：

import numpy as nm
x,y,z = nm.mgrid[[slice(dm) for dm in abc.shape]]