如何按字母顺序排序索引

时间:2018-02-14 15:32:01

标签: python sorting natural-sort

如何按字母顺序对pandas索引进行排序

idx = pd.Index(['A1 yes', 'A2 no', 'A3 no', 'A10 yes'])

idx.str[:3].to_series().value_counts().sort_index()

A1  1
A10  1
A3 1

如何将其分类为A1,A3和A10而不是A1,A10和A3?

1 个答案:

答案 0 :(得分:2)

使用natsorted + reindex

from natsort import natsorted

s = idx.str[:3].to_series().value_counts()
s = s.reindex(natsorted(s.index))
print (s)
A1     1
A2     1
A3     1
A10    1
dtype: int64

extract仅按数字排序:

s = s.iloc[s.index.str.extract('(\d+)', expand=False).astype(int).argsort()]
print (s)
A1     1
A2     1
A3     1
A10    1
dtype: int64

最后如果想按字符串用数字排序:

df = (s.index.to_series().str.extract('(?P<a>\D+)(?P<b>\d+)', expand=True)
      .assign(b=lambda x: x['b'].astype(int))
      .sort_values(['a','b']))
print (df)
     a   b
A1   A   1
A2   A   2
A3   A   3
A10  A  10

s = s.reindex(df.index)
print (s)
A1     1
A2     1
A3     1
A10    1
dtype: int64