如何将 Numpy 数组（行 x 列）转换为 XYZ 坐标数组？

Question

我有一个来自相机（灰度图像）的输入数组，看起来像：

[
[0.5, 0.75, 0.1, 0.6], 
[0.3, 0.75, 1.0, 0.9] 
]

实际尺寸 = 434x512

我需要一个 XYZ 坐标列表的输出：

即[[x,y,z],[x,y,z],...]

[[0,0,0.5],[1,0,0.75],[2,0,0.1],[3,0,0.6],[0,1,0.3],[1,1,0.75],[2,1,1.0],[3,1,0.9]]

有没有使用 Numpy 的有效方法？

Answer 1

这是一个方法 -

m,n = a.shape
R,C = np.mgrid[:m,:n]
out = np.column_stack((C.ravel(),R.ravel(), a.ravel()))

样本运行-

In [45]: a
Out[45]: 
array([[ 0.5 ,  0.75,  0.1 ,  0.6 ],
       [ 0.3 ,  0.75,  1.  ,  0.9 ]])

In [46]: m,n = a.shape
    ...: R,C = np.mgrid[:m,:n]
    ...: out = np.column_stack((C.ravel(),R.ravel(), a.ravel()))
    ...: 

In [47]: out
Out[47]: 
array([[ 0.  ,  0.  ,  0.5 ],
       [ 1.  ,  0.  ,  0.75],
       [ 2.  ,  0.  ,  0.1 ],
       [ 3.  ,  0.  ,  0.6 ],
       [ 0.  ,  1.  ,  0.3 ],
       [ 1.  ,  1.  ,  0.75],
       [ 2.  ,  1.  ,  1.  ],
       [ 3.  ,  1.  ,  0.9 ]])

In [48]: out.tolist() # Convert to list of lists if needed
Out[48]: 
[[0.0, 0.0, 0.5],
 [1.0, 0.0, 0.75],
 [2.0, 0.0, 0.1],
 [3.0, 0.0, 0.6],
 [0.0, 1.0, 0.3],
 [1.0, 1.0, 0.75],
 [2.0, 1.0, 1.0],
 [3.0, 1.0, 0.9]]

如何将 Numpy 数组（行 x 列）转换为 XYZ 坐标数组？

How to convert a Numpy array (Rows x Cols) to an array of XYZ coordinates?

numpy

data-conversion