的Python：指定列表多維數組元素

Question

我想要做一個分配給一個多維數組，其中數組的每個元素是3個短整型：

a = ndarray([3,3,3], dtype='u2,u2,u2') 
a[2,2,2] = [1,2,3] 

Traceback (most recent call last): File "", line 1, in a[2,2,2] = [1,2,3] TypeError: expected a readable buffer object

我將使用大數組，並希望直接對數組進行索引以獲得性能。在Python中做這件事的好方法是什麼？

感謝您瞭解如何做到這一點？

Answer 1

的dtype='u2,u2,u2'數組的元素是一個元組三個短褲，而不是一個列表三個短褲。所以：

a[2,2,2] = (1,2,3) 

（在括號當然是沒有必要的，但我用它們來使它明顯這是一個元組。）

你也可以通過它的array，如果你想：

a[2,2,2] = np.array([1,2,3]) 

當然，這裏的錯誤信息當然可能會更好......它實際上抱怨的是比你期望的更深的東西，它並沒有幫助你調試問題。