有沒有辦法讓numpy矩陣存儲對象？

Question

我想在numpy矩陣中存儲一個元組，但是它好像會返回一個錯誤。有沒有辦法去解決它？

>>> import numpy 
>>> y = numpy.zeros((4,4)) 
>>> y[1][1] = (1,1) 

ValueError: setting an array element with a sequence. 

感謝

Answer 1

使用dtype=object，你可以把任何東西你的數組中要：

>>> arr = np.zeros((4, 4), dtype=object) 
>>> arr[1, 1] = (1, 1) 
>>> arr 
array([[0, 0, 0, 0], 
     [0, (1, 1), 0, 0], 
     [0, 0, 0, 0], 
     [0, 0, 0, 0]], dtype=object) 

Answer 2

您可以通過使用numpy's structured arrays做到這一點。例如：

>>> import numpy as np 
>>> y = np.zeros((4, 4), dtype=("i8, i8")) 
array([[(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)]], 
     dtype=[('f0', '<i8'), ('f1', '<i8')]) 

>>> y[1,1] = (1,1) 
>>> y 
array([[(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (1L, 1L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)], 
     [(0L, 0L), (0L, 0L), (0L, 0L), (0L, 0L)]], 
     dtype=[('f0', '<i8'), ('f1', '<i8')])