在Numpy中,可以使用切片操作来为不同索引的数组设置每列的值。具体步骤如下:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
value = 10
index = [0, 2]
arr[:, index:] = value
arr[:, index:]
表示选择所有行和指定索引之后的列。=
表示将选定的列设置为指定的值。完整的代码示例如下:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
value = 10
index = [0, 2]
arr[:, index:] = value
print(arr)
输出结果为:
[[10 2 10]
[10 5 10]
[10 8 10]]
这样,我们就成功地为不同索引的数组设置了每列的值。在这个例子中,我们选择了索引为0和2的列,并将它们的值设置为10。
领取专属 10元无门槛券
手把手带您无忧上云