我可以使用列表来寻址numpy数组吗？

Question

我想给一个列表寻址的numpy数组的元素赋值。这可能吗？这似乎是你应该做的事情。你知道吗

我试过：

q = np.zeros((2,2,2))
index = [0,0,0]
print(index)


q[index]=4.3
print(q)

它没有给出一个错误，这是有希望的，但q现在是：

[[[ 4.3  4.3]
  [ 4.3  4.3]]

 [[ 0.   0. ]
  [ 0.   0. ]]]

相对于：

[[[ 4.3  0. ]
  [ 0.   0.]]

 [[ 0.   0. ]
  [ 0.   0. ]]]

就像我希望的那样。你知道吗

事先谢谢你的帮助。你知道吗

Answer 1

不能使用list索引单个元素-它必须是tuple：

import numpy as np

q = np.zeros((2,2,2))
index = [0,0,0]
print(index)


q[tuple(index)]=4.3
print(q)
[0, 0, 0]
[[[ 4.3  0. ]
  [ 0.   0. ]]

 [[ 0.   0. ]
  [ 0.   0. ]]]