计算 pandas 列中列表元素的总数

Question

我有一个熊猫数据 A 列 keywords 作为（这里我只显示 4 行，但实际上有数百万行）:-

  keywords
 ['loans','mercedez','bugatti']
 ['trump','usa']
 ['galaxy','7s','canon','macbook']
 ['beiber','spiderman','marvels','ironmen']

我想对列中的列表元素总数求和 keywords 并将其存储到某个变量中。就像是

total_sum=elements in keywords[0]+elements in keywords[1]+elements in
          keywords[2]+elements in keywords[3]

total_sum=3+2+4+4
total_sum=13

我如何在熊猫中做到这一点？

原文由 Learner 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

IIUC

设置

df = pd.DataFrame()
df['keywords']=[['loans','mercedez','bugatti'],
                ['trump','usa'],
                ['galaxy','7s','canon','macbook'],
                ['beiber','spiderman','marvels','ironmen']]

然后juse str.len 和 sum

 df.keywords.str.len().sum()

---

细节：

 df.keywords.str.len()

0    3
1    2
2    4
3    4
Name: keywords, dtype: int64

---

Ps：如果你有 strings 看起来像一个列表，使用 ast.literal_eval 转换为列表。

 df.keywords.transform(ast.literal_eval).str.len().sum()

原文由 rafaelc 发布，翻译遵循 CC BY-SA 4.0 许可协议