Как я могу отсортировать строки в файле данных pandas в соответствии со значением в определенном столбце? [дубликат]
Моя программа Python генерирует файл данных в формате pandas
Source LogP MolWt HBA HBD
0 cne_1 1.1732 263.405 3 1
1 cne_10 2.6639 197.237 2 0
2 cne_100 -0.2886 170.193 4 2
3 cne_1000 1.9644 304.709 5 1
4 cne_1001 1.4986 162.144 3 1
... ... ... ... ... ...
1031 cne_995 3.0179 347.219 4 2
1032 cne_996 4.8419 407.495 6 2
1033 cne_997 3.3560 354.524 3 1
1034 cne_998 7.5465 635.316 4 2
1035 cne_999 3.3514 389.556 4 1
Мне нужно отсортировать строки на основе второго (исходного) столбца по номеру, поэтому правильный порядок строк после сортировки должен быть: cne_1, cne_2, cne_3, cne_4 и т. Д. Я пытался использовать:
df_sorted = df.sort_values('Source', ascending=True)
но это не произвело никаких изменений в порядке строк.
Ответы
Для последних версий pandas возможно использование параметра key
с разделением значений по _
и преобразованием значений в целые числа:
df_sorted = df.sort_values('Source', key=lambda x: x.str.split('_').str[1].astype(int))
Или можно получить позиции отсортированных значений Series.argsortи перейти к DataFrame.iloc:
df_sorted = df.iloc[df['Source'].str.split('_').str[1].astype(int).argsort()]
print (df_sorted)
Source LogP MolWt HBA HBD
0 cne_1 1.1732 263.405 3 1
1 cne_10 2.6639 197.237 2 0
2 cne_100 -0.2886 170.193 4 2
1031 cne_995 3.0179 347.219 4 2
1032 cne_996 4.8419 407.495 6 2
1033 cne_997 3.3560 354.524 3 1
1034 cne_998 7.5465 635.316 4 2
1035 cne_999 3.3514 389.556 4 1
3 cne_1000 1.9644 304.709 5 1
4 cne_1001 1.4986 162.144 3 1
Получите целочисленное значение в столбце, а затем выполните сортировку по нему.
df['sortIndex'] = df.Source.str.replace('cne_', '', regex=False).astype(int)
df_sorted = df.sort_values('sortIndex', ascending=True)
Извлеките цифры, преобразуйте их в int и отсортируйте соответственно. ( .sort_values(0)
поскольку безымянный столбец получает имя 0
автоматически)
df_sorted = df.loc[df["Source"].str.extract(r"_(\d+)").astype(int).sort_values(0).index]
Результат
print(df_sorted)
Source LogP MolWt HBA HBD
0 cne_1 1.1732 263.405 3 1
1 cne_10 2.6639 197.237 2 0
2 cne_100 -0.2886 170.193 4 2
1031 cne_995 3.0179 347.219 4 2
1032 cne_996 4.8419 407.495 6 2
1033 cne_997 3.3560 354.524 3 1
1034 cne_998 7.5465 635.316 4 2
1035 cne_999 3.3514 389.556 4 1
3 cne_1000 1.9644 304.709 5 1
4 cne_1001 1.4986 162.144 3 1