可以使用numpy的广播功能来实现不同形状的数组相加。广播是一种numpy中的特性,它可以自动地处理不同形状的数组的运算。
下面是一个示例代码,展示了如何使用广播功能来相加不同形状的numpy数组:
import numpy as np
# 创建两个不同形状的数组
a = np.array([[1, 2, 3], [4, 5, 6]]) # 形状为 (2, 3)
b = np.array([1, 2, 3]) # 形状为 (3,)
# 使用广播相加两个数组
result = a + b
print(result)
输出结果:
[[2 4 6]
[5 7 9]]
在这个示例中,数组a
的形状是(2, 3),数组b
的形状是(3,)。由于数组b
的维度较低,它会被扩展为与a
相同的形状(2, 3)。然后,numpy会自动地对两个数组进行元素级的相加运算,得到最终的结果。
这个示例展示了如何使用广播功能来相加不同形状的numpy数组。你可以根据自己的需求,调整数组的形状,并使用广播来进行相加运算。
上一篇:不同形状的列表