在numpy中,一个16x2的数组,每四行求和,输出成一个4x2的数组
时间: 2024-01-26 21:03:53 浏览: 105
可以使用numpy中的reshape和sum函数来实现这个操作。具体方式如下:
```python
import numpy as np
# 创建一个16x2的数组
arr = np.random.rand(16, 2)
# 将数组按照每4行进行分组,得到4个4x2的小数组
sub_arrays = arr.reshape(4, 4, 2)
# 对每个小数组进行求和操作,得到4个2维向量
sums = sub_arrays.sum(axis=1)
# 输出结果
print(sums)
```
输出结果为一个4x2的数组,其中每个元素表示对应列上4行元素的和:
```
[[2.58076527 1.44201021]
[1.95071284 2.34387655]
[2.40964811 1.81177069]
[2.26590951 2.10003366]]
```
阅读全文