我有一个df
review review_id word_count
['abc def', 'ghi'] [1, 2] [2, 1]
['jkl mno pqr', 'stu', 'vwx'] [3, 4, 5] [3, 1, 1]
对于df中的每一行，我想为review和review\u id创建新的列，其中只包含最大字数的review/review\u id.
review review_id
abc def 1
jkl mno pqr 3
我只能想得到word\u count中最大值的索引.欢迎任何建议.谢谢
使用DataFrame.explode比较每组的最大值(从行中)，如果需要所有最大值，则按boolean indexing Select 行:
df = df.explode(['review','review_id','word_count'])
#for oldier pandas versions
#df = df.apply(lambda x: x.explode())
df = df[df['word_count'].eq(df.groupby(level=0)['word_count'].transform('max'))]
print (df)
review review_id word_count
0 abc def 1 2
1 jkl mno pqr 3 3