指定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))]) 

我不得不多想一些，弄清楚到底發生了什麼。