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

Question

我想为列表寻址的 numpy 数组的元素分配一个值。这可能吗？这似乎是你应该能够做的事情。

我尝试过:

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


q[index]=4.3
print(q)

这没有给出错误，这是有希望的，但现在是:

[[[ 4.3  4.3]
  [ 4.3  4.3]]

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

相对于:

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

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

正如我所希望的那样。

预先感谢您的帮助。

Answer 1

您不能使用列表来索引单个元素 - 它必须是元组:

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. ]]]