如果有人可以提供帮助,我会非常感激,因为它使我疯狂了几个小时!
我有一个ndarray如下:
array([[[0, 0],
[0, 2],
[0, 4]],
[[1, 0],
[1, 2],
[1, 4]]])
我想将其转换为记录数组:
array([[(0, 0),
(0, 2),
(0, 4)],
[(1, 0),
(1, 2),
(1, 4)]],
dtype=[('x', '<i4'), ('y', '<i4')])
您可以简单地使用.view
“更改” dtype和.reshape
“删除”最后一个尺寸:
>>> import numpy as np
>>> arr = np.array([[[0, 0], [0, 2], [0, 4]],
... [[1, 0], [1, 2], [1, 4]]])
>>> newarr = arr.view(dtype=np.dtype([('x', 'i4'), ('y', 'i4')]))
>>> newarr = newarr.reshape(newarr.shape[:-1])
>>> newarr
array([[(0, 0), (0, 2), (0, 4)],
[(1, 0), (1, 2), (1, 4)]],
dtype=[('x', '<i4'), ('y', '<i4')])
请注意,您必须注意,.view
在此处硬编码dtype可能不是“安全的”。但是您可以简单地重用当前的dtype:
>>> newarr = arr.view(dtype=np.dtype([('x', arr.dtype), ('y', arr.dtype)]))
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句