NumPy的广播机制可以让不同维度的数组进行运算,而不需要手动扩展数组的维度。广播机制的规则如下:
下面是一个简单的例子,展示了如何使用广播机制进行数组的运算:
import numpy as np
# 创建两个数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 使用广播机制进行数组运算
result = a + b
print(result)
在上面的例子中,数组a和数组b的维度不同,但是通过广播机制,数组b被扩展为[4, 5, 6],然后和数组a进行元素级的加法运算。最终得到的结果是[5, 7, 9]。