python - 如何计算跨列列表中元素的平均值？

Question

假设我有一个列表数组

B = [[1,2,3],[1,2,3,4],[1,2]]

我想计算各列列表中元素的平均值。我该怎么做？

即

如何获得等于最长列表的最终平均值数组:

[(1+1+1)/3,(2+2+2)/3,(3+3)/2,4/1] = [1,2,3,4]

我试过:

final_array = np.array([mean(a) for a in zip(*(B))])

但这只给我一个与我的最短列表一样长的数组。这是口罩派上用场了吗？如果列表数组让您感到畏缩，我深表歉意，我仍在适应 Python。

Answer 1

你可以使用 pandas 的 DataFrame。

from pandas import DataFrame

B = [[1,2,3],[1,2,3,4],[1,2]]
df = DataFrame(B)
df.mean(axis=0)
""""
df
   0  1   2   3
0  1  2   3 NaN
1  1  2   3   4
2  1  2 NaN NaN

df.mean(axis=0)
0    1
1    2
2    3
3    4
"""