可以使用 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]]
numpy 某列减去一个固定值

原文地址: https://www.cveoy.top/t/topic/bgbc 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录