我正在寻找一个 numpy 函数,它的作用相当于:
indices = set([1, 4, 5, 6, 7])
zero = numpy.zeros(10)
for i in indices:
zero[i] = 42
最佳答案
你可以只给它一个索引列表:
indices = [1, 4, 5, 6, 7]
zero = numpy.zeros(10)
zero[indices] = 42
关于python - numpy 函数将数组元素设置为给定索引列表的值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/8373079/