如何对NumPy数组的指定列（第1列与第4列）进行求和操作

首先需要注意：NumPy 采用0起始索引规则，你可以根据自己对“第1列”“第4列”的计数习惯选择对应的索引：

• 如果是0起始计数：最左侧第1列索引为0，第4列索引为3
• 如果是日常1起始计数：最左侧第1列索引为1，第4列索引为4

你可以通过花式索引提取目标列后，按需求调用求和方法即可，示例代码如下：

import numpy as np
a = np.arange(24).reshape(4,6)

# 示例：取1起始计数的第1列、第4列，如需取0起始的把索引改成[0,3]即可
target_cols = a[:, [1, 4]]

# 1. 计算两列所有元素的总求和
total = target_cols.sum()
# 输出结果：92

# 2. 计算每行中两列元素的和（输出长度为4的数组，对应4行的结果）
row_wise_sum = target_cols.sum(axis=1)
# 输出结果：array([ 5, 17, 29, 41])

# 3. 分别计算两列各自的总和（输出长度为2的数组，对应两列的结果）
col_wise_sum = target_cols.sum(axis=0)
# 输出结果：array([40, 52])

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