python计算二维np.array某一列的和
时间: 2024-02-17 09:24:36 浏览: 261
python 实现二进制求和
要计算二维 NumPy 数组中某一列的和,你可以使用 `numpy.sum()` 函数,并指定 `axis` 参数为 0。这样可以按列进行求和。
以下是一个示例代码:
```python
import numpy as np
# 创建示例二维数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 计算第一列的和
column_sum = np.sum(arr[:, 0])
print(column_sum) # 输出: 12
```
在上述代码中,我们创建了一个示例的二维数组 `arr`。然后,使用 `np.sum()` 函数计算了第一列的和,通过切片 `arr[:, 0]` 来选择第一列。最后,将结果赋值给变量 `column_sum` 并打印出来。
如果你想计算所有列的和,可以省略切片中的索引,直接使用 `np.sum(arr, axis=0)`。
希望这个示例能够帮助你计算二维 NumPy 数组中某一列的和!如果还有其他问题,请随时提问。
阅读全文