使用列表访问数组

Question

我有一个数组A和一个列表b。我想将数字c放在数组A的位置b中。例如，假设

import numpy
size = 4
size_maker = [2] * size
A = numpy.zeros(shape=size_maker)

b = [0,1,0,0]
c = 20

我想要A [0,1,0,0] = 20.任何想法？

Answer 1

您只需要使用元组而不是列表。

A[tuple(b)] = c