如何对二维数组第二列的每个元素减去其平方根？

需求：对二维数组的第二列每个值减去其平方根，将以下原数组：

array([[ 0,  9],
       [ 2,  16],
       [ 4,  25],
       [10, 36]])

转换为目标数组：

array([[ 0,  6],
       [ 2,  12],
       [ 4,  20],
       [10, 30]])

你尝试的代码B[:, 1] -= np.sqrt(B[:, 1])完全可以实现需求，完整执行流程如下：

1. 导入NumPy库

import numpy as np

2. 定义原数组

B = np.array([[ 0,  9],
              [ 2,  16],
              [ 4,  25],
              [10, 36]])

3. 执行第二列的数值处理

B[:, 1] -= np.sqrt(B[:, 1])

4. 查看处理后的数组

print(B)

代码说明 #

• B[:, 1]：选取数组B的第二列（NumPy数组索引从0开始）
• np.sqrt(B[:, 1])：计算第二列每个元素的平方根
• -=：原地减法赋值，直接将第二列元素更新为“原值减自身平方根”的结果

内容的提问来源于stack exchange，提问作者Crimzon Fiasco