numpy的广播功能举例
时间: 2023-06-03 07:01:43 浏览: 100
非常感谢您的提问!关于numpy的广播功能,它是指在numpy中,对于形状不同的数组进行某些操作时,numpy会自动地对其中维度较小的数组进行“广播”,实现相同形状的数组间运算,提高运算效率。下面是一个举例:
```
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 对两个形状不同的数组进行加法运算
c = a + b
print(c) # 输出 [5 7 9]
```
在上面的例子中,由于a和b都是一维数组,且长度相同,因此numpy会将它们看作同一个形状,进行相应的运算。
希望以上回答对您有所帮助,如有不清楚的地方,还请随时提问!
阅读全文