Início / coding / Perguntas / 78915880
Accepted
Jim Macaulay
Asked: 2024-08-27 02:34:38 +0800 CST

Transpor uma coluna no pandas DataFrame

  • 772

Tenho um dataframe abaixo, estou tentando transpor os dados com base na coluna Local. Para cada lista de valores na coluna Local, preciso gerar cada linha.

Language   Capital   Place
Tamil      Chennai   ['Chennai', 'Vellore', 'Trichy', 'Madurai']
Kerala     Kochi     ['Kochi', 'Trivandrum']

Resultado esperado

Language     Capital     Place
Tamil        Chennai     Chennai
Tamil        Chennai     Vellore
Tamil        Chennai     Trichy
Tamil        Chennai     Madurai 
Kerala       Kochi       Kochi
Kerala       Kochi       Trivandrum

Eu tentei de várias maneiras, usando pandas transpose, sem conseguir o resultado esperado. Também recuperei e converti a coluna Place em uma série de dataframe, ainda não consegui obter o resultado.

Qualquer sugestão será apreciada

python

2 respostas

  1. Best Answer

    Use ast.literal_eval para converter representações de string em listas reais:

    import ast
df['Place'] = df['Place'].apply(ast.literal_eval)

    Transforme cada elemento da lista em 'Place' em uma linha separada, replique os outros valores da coluna e redefina o índice do DataFrame para limpeza:

    df_exploded = df.explode('Place').reset_index(drop=True)
print(df_exploded)
    • 2

  2. Você pode conseguir isso usando a função explode() da biblioteca pandas, que transforma cada elemento de uma coluna semelhante a uma lista em uma linha, replicando os outros valores da coluna.

    import pandas as pd

data = {
    'Language': ['Tamil', 'Kerala'],
    'Capital': ['Chennai', 'Kochi'],
    'Place': [['Chennai', 'Vellore', 'Trichy', 'Madurai'], ['Kochi', 'Trivandrum']]
}
df = pd.DataFrame(data)
df_exploded = df.explode('Place')
print(df_exploded)
    • 0

relate perguntas