python - 如何对包含数组的列求和

Question

我在总结每个单元格中包含数组的数据帧的列时遇到问题。 我尝试使用 df.sum() 汇总列，期望获得总列数组，例如“common”列的 [4,1,1,4,1]。但我只有一个空系列。

df_sum = df.sum()
print(df_sum)

Series([], dtype: float64)

在这种情况下如何获取汇总列？

Answer 1

嗯，在 pandas DataFrames 中使用 object dtypes 通常不是一个好主意，特别是用 python 列表填充单元格，因为你会损失性能。

尽管如此，您可以通过使用 itertools.chain.from_iterable 来完成此操作

df.apply(lambda s: list(it.chain.from_iterable(s.dropna())))

你也可以使用sum，但我想说它比较慢

df.apply(lambda s: s.dropna().sum())

我可以理解为什么您认为 df.sum 可以在这里工作，甚至显式设置 skipna=True，但是矢量化的 df.sum code> 在这种情况下表现出奇怪的行为。但话又说回来，这些是使用包含列表的 DataFrame 的缺点