python - Pandas:如何根据特定的后缀值对行进行排序？

Question

我的 Pandas 数据框包含从 csv 文件读取的以下数据:

id,values
 1001-MAC,     10
 1034-WIN,     20
 2001-WIN,     15
 3001-MAC,     45
 4001-LINUX,   12
 4001-MAC,     67

df = pd.read_csv('example.csv')
df.set_index('id', inplace=True)

我必须根据给定的后缀列表 = [“WIN”、“MAC”、“LINUX”] 的 id 列顺序对这个数据框进行排序。因此，我想得到以下输出:

 id,values
 1034-WIN,     20
 2001-WIN,     15
 1001-MAC,     10
 3001-MAC,     45
 4001-MAC,     67
 4001-LINUX,   12
 

我该怎么做？

Answer 1

这是一种方法:

import pandas as pd
df = pd.read_csv('example.csv')
idx = df.id.str.split('-').str[1].sort_values(ascending=False).index

df = df.loc[idx]
df.set_index('id', inplace=True)
print(df)