gpt4 book ai didi

python - 在包含最长列表的 Pandas DF 中查找列的名称

转载 作者:太空狗 更新时间:2023-10-29 20:36:23 25 4
gpt4 key购买 nike

给定一个 Pandas DataFrame,其列表存储在多个列中,是否有一种简单的方法可以找到包含每行最长列表的列名?

例如,使用此数据:

                          positive                 negative          neutral
1 [marvel, moral, bold, destiny] [] [view, should]
2 [beautiful] [complicated, need] []
3 [celebrate] [crippling, addiction] [big]

我想将“正”标识为第 1 行的最长列表列,将“负”标识为第 2 行和第 3 行。

我想我可以使用 str.len() 来计算列表长度并使用 idmax() 来获取列名,但不知道如何将它们结合起来。

最佳答案

IIUC:

In [227]: df.applymap(len).idxmax(axis=1)
Out[227]:
0 positive
1 negative
2 negative
dtype: object

关于python - 在包含最长列表的 Pandas DF 中查找列的名称,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45827199/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com