numpy 某列减去一个固定值
可以使用 numpy 的广播功能将固定值广播到整个数组中,并执行减法操作。
例如,假设有一个数组 arr,想要将第二列减去 5:
import numpy as np
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
arr[:, 1] -= 5
这里使用了数组切片语法 [:, 1],表示选择所有行和第二列。然后将得到的切片减去 5。注意,这里使用的是原地操作符 -=,它会直接修改原始数组。
结果如下所示:
[[ 1 -3 3]
[ 4 0 6]
[ 7 3 9]]
原文地址: https://www.cveoy.top/t/topic/bgbc 著作权归作者所有。请勿转载和采集!