将ndarray的最后一个维度转换为元组/重新排列

时间:2017-04-15 12:25:44

标签: python arrays numpy

如果有人可以帮忙解决这个问题,我会非常感激,因为它让我有点疯狂了几个小时!

我有一个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')])

1 个答案:

答案 0 :(得分:3)

您只需使用.view即可更改&#34; dtype和.reshape到&#34;删除&#34;最后一个维度:

>>> 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可能不是&#34;安全&#34;这里。但您可以简单地重用当前的dtype:

>>> newarr = arr.view(dtype=np.dtype([('x', arr.dtype), ('y', arr.dtype)]))