Como posso classificar as strings no arquivo de dados do pandas de acordo com o valor na coluna específica? [duplicado]
Meu programa Python gera um arquivo de dados no formato 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
Preciso classificar as strings com base na segunda coluna (Origem) de acordo com o número, então a ordem correta das linhas após a classificação deve ser: cne_1, cne_2, cne_3, cne_4 et tentei usar:
df_sorted = df.sort_values('Source', ascending=True)
mas não provocou nenhuma mudança na ordem das linhas.
Respostas
Para as últimas versões do pandas é possível usar o parâmetro key
com valores divididos _
e convertendo os valores para inteiros:
df_sorted = df.sort_values('Source', key=lambda x: x.str.split('_').str[1].astype(int))
Ou é possível obter posições de valores classificados por Series.argsorte passar para 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
Obtenha o valor inteiro na coluna e classifique usando isso.
df['sortIndex'] = df.Source.str.replace('cne_', '', regex=False).astype(int)
df_sorted = df.sort_values('sortIndex', ascending=True)
Extraia os dígitos, converta em int e classifique de acordo. ( .sort_values(0)
porque a coluna sem nome é nomeada 0
automaticamente)
df_sorted = df.loc[df["Source"].str.extract(r"_(\d+)").astype(int).sort_values(0).index]
Resultado
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