使用numpy库中的array slicing来实现。
import numpy as np
# 创建一个3行3列的矩阵
matrix = np.array([[1,2,3],[4,5,6],[7,8,9]])
# 将矩阵的第0列和第2列的值加上1
matrix[:,[0,2]] += 1
print(matrix)
输出:
[[ 2 2 4]
[ 5 5 7]
[ 8 8 10]]
在上述代码中,:
表示要对整个矩阵进行操作。[0,2]
表示要将第0列和第2列的值加上1。使用+=
可以将指定的值直接加到对应位置上。