按字符串长度对数据帧进行排序
Sort dataframe by string length
我想按名称长度排序。 sort_values
似乎没有 key
参数,所以我不确定如何完成此操作。这是一个测试 df:
import pandas as pd
df = pd.DataFrame({'name': ['Steve', 'Al', 'Markus', 'Greg'], 'score': [2, 4, 2, 3]})
您可以使用 reindex
of index
of Series
created by len
with sort_values
:
print (df.name.str.len())
0 5
1 2
2 6
3 4
Name: name, dtype: int64
print (df.name.str.len().sort_values())
1 2
3 4
0 5
2 6
Name: name, dtype: int64
s = df.name.str.len().sort_values().index
print (s)
Int64Index([1, 3, 0, 2], dtype='int64')
print (df.reindex(s))
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2
df1 = df.reindex(s)
df1 = df1.reset_index(drop=True)
print (df1)
name score
0 Al 4
1 Greg 3
2 Steve 2
3 Markus 2
我发现此解决方案更直观,特别是如果您想稍后根据列长度执行某些操作。
df['length'] = df['name'].str.len()
df.sort_values('length', ascending=False, inplace=True)
现在您的数据框将有一个名称为 length
的列,其中包含列 name
中的字符串长度值,并且整个数据框将按降序排序。
@jezrael 的回答很好,解释的也很好。这是最终结果:
index_sorted = df.name.str.len().sort_values(ascending=True).index
df_sorted = df.reindex(index_sorted)
df_sorted = df_sorted.reset_index(drop=True)
一个奇特的最小解决方案:
df.iloc[df.agg({"name":len}).sort_values('name').index]
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2
将 DataFrame.sort_values
与 key
参数一起使用
因为 pandas >= 1.1.0
:
我们现在可以在 sort_values
方法中传递字符串的长度或任何其他自定义键:
df = pd.DataFrame({
'name': ['Steve', 'Al', 'Markus', 'Greg'],
'score': [2, 4, 2, 3]
})
print(df)
name score
0 Steve 2
1 Al 4
2 Markus 2
3 Greg 3
df.sort_values(by="name", key=lambda x: x.str.len())
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2
我想按名称长度排序。 sort_values
似乎没有 key
参数,所以我不确定如何完成此操作。这是一个测试 df:
import pandas as pd
df = pd.DataFrame({'name': ['Steve', 'Al', 'Markus', 'Greg'], 'score': [2, 4, 2, 3]})
您可以使用 reindex
of index
of Series
created by len
with sort_values
:
print (df.name.str.len())
0 5
1 2
2 6
3 4
Name: name, dtype: int64
print (df.name.str.len().sort_values())
1 2
3 4
0 5
2 6
Name: name, dtype: int64
s = df.name.str.len().sort_values().index
print (s)
Int64Index([1, 3, 0, 2], dtype='int64')
print (df.reindex(s))
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2
df1 = df.reindex(s)
df1 = df1.reset_index(drop=True)
print (df1)
name score
0 Al 4
1 Greg 3
2 Steve 2
3 Markus 2
我发现此解决方案更直观,特别是如果您想稍后根据列长度执行某些操作。
df['length'] = df['name'].str.len()
df.sort_values('length', ascending=False, inplace=True)
现在您的数据框将有一个名称为 length
的列,其中包含列 name
中的字符串长度值,并且整个数据框将按降序排序。
@jezrael 的回答很好,解释的也很好。这是最终结果:
index_sorted = df.name.str.len().sort_values(ascending=True).index
df_sorted = df.reindex(index_sorted)
df_sorted = df_sorted.reset_index(drop=True)
一个奇特的最小解决方案:
df.iloc[df.agg({"name":len}).sort_values('name').index]
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2
将 DataFrame.sort_values
与 key
参数一起使用
因为 pandas >= 1.1.0
:
我们现在可以在 sort_values
方法中传递字符串的长度或任何其他自定义键:
df = pd.DataFrame({
'name': ['Steve', 'Al', 'Markus', 'Greg'],
'score': [2, 4, 2, 3]
})
print(df)
name score
0 Steve 2
1 Al 4
2 Markus 2
3 Greg 3
df.sort_values(by="name", key=lambda x: x.str.len())
name score
1 Al 4
3 Greg 3
0 Steve 2
2 Markus 2