Python中是否有计算两个列表元素两两均值(或求和等)的更优方法?
更Pythonic的两两元素均值计算方案(基于NumPy)
你可以利用NumPy的广播机制来实现,完全避免显式嵌套循环,代码简洁且效率更高:
import numpy as np A = np.array([1, 2, 3]) B = np.array([8, 12, 11]) # 通过广播实现两两元素的均值计算 C = (A[:, None] + B) / 2 print(C)
运行结果和你之前的完全一致:
array([[4.5, 6.5, 6. ], [5. , 7. , 6.5], [5.5, 7.5, 7. ]])
核心思路
A[:, None]将一维数组A转换为列向量(形状变为(3,1)),触发NumPy的广播机制,自动将A和B扩展为相同的(3,3)形状- 直接对扩展后的数组进行逐元素运算,最后除以2得到均值,比逐个调用
np.mean更高效 - 这种方式完全符合Pythonic的简洁原则,同时利用NumPy的底层优化,处理大规模数组时性能优势会更明显
如果需要实现其他两两运算(比如求和、乘积),只需要调整运算逻辑即可:
- 两两求和:
A[:, None] + B - 两两乘积:
A[:, None] * B
内容的提问来源于stack exchange,提问作者db_
相关产品推荐
相关产品推荐

