You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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_

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.14 03:01:09