python二维数组的列向量求和,得到一个一维向量
时间: 2023-12-20 18:49:31 浏览: 157
python+numpy按行求一个二维数组的最大值方法
可以使用numpy库中的sum函数来实现。
假设二维数组为arr,可以先使用transpose函数将其转置为列向量,再使用sum函数对每列进行求和,最终得到一个一维向量,代码如下:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col_sum = np.sum(arr.T, axis=1)
print(col_sum)
```
输出结果为:
```
[12 15 18]
```
其中,T表示转置操作,axis=1表示对每列进行求和。
阅读全文