如何使用numpy的broadcast函数来实现shape的自动匹配,将(2,)改为(2, 2)
时间: 2023-06-07 17:06:45 浏览: 169
python numpy np.arctan2()函数(批量计算反正切?)
5星 · 资源好评率100%
可以使用numpy的reshape函数来实现shape的自动匹配,代码如下:
```python
import numpy as np
# 将(2,)改为(2, 2)
a = np.array([1, 2])
b = np.broadcast_to(a.reshape(2, 1), (2, 2))
```
通过将a reshape为(2, 1),再通过broadcast_to函数将其扩展为(2, 2)的矩阵b,完成了shape的自动匹配。
阅读全文