python - 按列表大小而不是 A-Z 对 Pandas 数据框进行排序

Question

我有一个如下所示的数据框:

username          group
user1             [A]
user2             [B, C]
user3             [A, B, C]
user4             [A, B]

我希望结果看起来像这样:

username          group
user3             [A, B, C]
user4             [A, B]
user2             [B, C]
user1             [A]

我查阅了 sort_values 的文档，它似乎主要涵盖 0-9 或 A-Z 排序。提前致谢

Answer 1

一种方法是创建一个包含长度的新列，然后按其排序:

df['len'] = df['group'].str.len()
df.sort_values(by='len', ascending=False).drop(columns='len')

输出是:

    group       
2   (A, B, C)   
1   (B, C)      
3   (A, B)      
0   (A,)