3D数组中每个元素的Numpy索引值
我有一个使用numpy mgrid命令创建的3D数组，以便每个元素都有一个确定的值，并且索引保留空间信息。 例如，如果在z轴上进行求和（第三维），则可以在matplotlib中使用生成的二维数组和imshow（）函数来获取具有不同像素值的图像。
我的问题是：如何获得此网格中每个元素的索引值（a，b，c）？
我需要使用索引值来计算每个点与网格原点的相对角度。 （例如theta = sin-1（sqrt（x ^ 2 + y ^ 2）/ sqrt（x ^ 2 + y ^ 2 + z ^ 2）））
也许这可以转换为另一个3D网格，其中每个元素是数组[a，b，c]？
我的意思并不完全清楚，但如果您正在寻找包含索引x，y和z的3d数组，那么以下内容可能适合您的需求; 假设您的数据保存在名为“abc”的3D数组中：
import numpy as nm
x,y,z = nm.mgrid[[slice(dm) for dm in abc.shape]]