视图的recarray作为ndarray

时间:2018-02-24 22:24:21

标签: python python-2.7 numpy numpy-ndarray

在以前的numpy版本中,我能够做到这样的事情:

data = np.array([(0,1,2),(3,4,5),(6,7,8)], dtype=np.dtype([('a','<f8'), ('b','<f8'),('c','<f8')]))


data_nd = data[['a','b']].view(('<f8',2))

这会让我看到原始的重新排列data作为np.float32类型的二维ndarray。

在numpy版本1.14中,当我尝试上面的第二行时,我得到错误:

ValueError: Changing the dtype to a subarray type is only supported if the total itemsize is unchanged

这是因为data[['a','b']]的项目大小仍然是24.好的,这没关系,但有没有办法获得更新项目大小的那些数据列的副本,以便我可以查看我的视图要什么?

例如,即使制作data[['a','b']]的副本(或深拷贝)仍然会产生24项,我不明白。

1 个答案:

答案 0 :(得分:0)

你是否意识到没有元组你的重新排列是3x3的复制字段值:

In [46]: data
Out[46]: 
array([[(0., 0., 0.), (1., 1., 1.), (2., 2., 2.)],
       [(3., 3., 3.), (4., 4., 4.), (5., 5., 5.)],
       [(6., 6., 6.), (7., 7., 7.), (8., 8., 8.)]],
      dtype=[('a', '<f8'), ('b', '<f8'), ('c', '<f8')])

无论如何,整个数组的视图仍然有效,你可以使用传统的列切片:

In [47]: data.view(('float',(3,)))
Out[47]: 
array([[[0., 0., 0.],
        [1., 1., 1.],
        [2., 2., 2.]],

       [[3., 3., 3.],
        [4., 4., 4.],
        [5., 5., 5.]],

       [[6., 6., 6.],
        [7., 7., 7.],
        [8., 8., 8.]]])
In [48]: _.shape
Out[48]: (3, 3, 3)

tolist是转换为结构化数组/从结构化数组转换的旧备用数据库。它仍然有效:

In [49]: np.array(data[['a','b']].tolist())
Out[49]: 
array([[[0., 0.],
        [1., 1.],
        [2., 2.]],

       [[3., 3.],
        [4., 4.],
        [5., 5.]],

       [[6., 6.],
        [7., 7.],
        [8., 8.]]])

data创建更改为包含元组,实际上并不会改变视图行为,但可能更接近您想要的(或者我想要的):

In [50]: data1 = np.array([(0,1,2),(3,4,5),(6,7,8)], dtype=np.dtype([('a','<f8')
    ...: , ('b','<f8'),('c','<f8')]))
In [51]: data1
Out[51]: 
array([(0., 1., 2.), (3., 4., 5.), (6., 7., 8.)],
      dtype=[('a', '<f8'), ('b', '<f8'), ('c', '<f8')])

请注意,子字段显示现在包含偏移参数。

In [53]: data1[['a','b']]
Out[53]: 
array([(0., 1.), (3., 4.), (6., 7.)],
      dtype={'names':['a','b'], 'formats':['<f8','<f8'], 'offsets':[0,8], 'itemsize':24})

底层数据库是相同的,因此3列视图可以使用和不使用子字段索引:

In [54]: data1[['a','b']].view((float,(3,)))
Out[54]: 
array([[0., 1., 2.],
       [3., 4., 5.],
       [6., 7., 8.]])
In [55]: data1.view((float,(3,)))
Out[55]: 
array([[0., 1., 2.],
       [3., 4., 5.],
       [6., 7., 8.]])

我还没有看过最新的变化。相反,我正在使用过去曾有过的各种技巧。转换为结构化数组或从结构化数组转换从未如此简单和万无一失。