python - 将列表的元素添加到某些列中的所有元素

Question

这是我的“问题”:给定一个 numpy 数组作为

array([[4, 3, 5, 1],
       [2, 3, 3, 4],
       [4, 2, 2, 7]])

我现在想做的是使用值列表

values = [3, 2]

并将它们添加到某些列的所有元素中，例如第 2 列和第 3 列导致

new_array([[4,6,7,1],
           [2,6,5,4],
           [4,5,4,7]])

如何做到这一点？

Answer 1

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

values = [3, 2]

a[:,(1,2)] += values

print(a)

打印:

[[4 6 7 1]
 [2 6 5 4]
 [4 5 4 7]]