numpy 结构化数组的分配问题

Question

我正在尝试在 numpy 中将代码分配给结构化数组的这一简单行，我不确定，但是当我将矩阵分配给我创建的结构化数组中的 sub_array 时会发生错误，如下所示：

new_type = np.dtype('a3,(2,2)u2')
x = np.zeros(5,dtype=new_type)
x[1]['f1'] = np.array([[1,1],[1,1]])
print x
Out[143]: 
array([('', [[0, 0], [0, 0]]), ('', [[1, 0], [0, 0]]),
   ('', [[0, 0], [0, 0]]), ('', [[0, 0], [0, 0]]),
   ('', [[0, 0], [0, 0]])], 
  dtype=[('f0', '|S3'), ('f1', '<u2', (2, 2))])

在这个阶段子数组的第二个字段不应该等于吗

[[1,1],[1,1]]

Answer 1

我认为您希望设置稍有不同。试试：

x['f1'][1] = np.array([[1,1],[1,1]])

导致：

In [43]: x = np.zeros(5,dtype=new_type)

In [44]: x['f1'][1] = np.array([[1,1],[1,1]])

In [45]: x
Out[45]: 
array([('', [[0, 0], [0, 0]]), ('', [[1, 1], [1, 1]]),
       ('', [[0, 0], [0, 0]]), ('', [[0, 0], [0, 0]]),
       ('', [[0, 0], [0, 0]])], 
      dtype=[('f0', '|S3'), ('f1', '<u2', (2, 2))])

这并不是说这不是奇怪的行为，因为 x['f1'][1] 和 x[1]['f1'] 打印的结果相同，但显然不同：

In [51]: x['f1'][1]
Out[51]: 
array([[1, 1],
       [1, 1]], dtype=uint16)

In [52]: x[1]['f1'] 
Out[52]: 
array([[1, 1],
       [1, 1]], dtype=uint16)

In [53]: x[1]['f1'] = 2

In [54]: x
Out[54]: 
array([('', [[0, 0], [0, 0]]), ('', [[2, 1], [1, 1]]),
       ('', [[0, 0], [0, 0]]), ('', [[0, 0], [0, 0]]),
       ('', [[0, 0], [0, 0]])], 
      dtype=[('f0', '|S3'), ('f1', '<u2', (2, 2))])

In [55]: x['f1'][1] = 3

In [56]: x
Out[56]: 
array([('', [[0, 0], [0, 0]]), ('', [[3, 3], [3, 3]]),
       ('', [[0, 0], [0, 0]]), ('', [[0, 0], [0, 0]]),
       ('', [[0, 0], [0, 0]])], 
      dtype=[('f0', '|S3'), ('f1', '<u2', (2, 2))])

我必须多考虑一下才能弄清楚到底发生了什么。