Como posso classificar as strings no arquivo de dados do pandas de acordo com o valor na coluna específica? [duplicado]

Dec 02 2020

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

5 jezrael Dec 02 2020 at 19:10

Para as últimas versões do pandas é possível usar o parâmetro keycom 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
3 Vishnudev Dec 02 2020 at 19:10

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)
2 BillHuang Dec 02 2020 at 19:10

Extraia os dígitos, converta em int e classifique de acordo. ( .sort_values(0)porque a coluna sem nome é nomeada 0automaticamente)

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