Return in a DataFrame - Python

Question

Return in a DataFrame - Python

Navigation

#1 by (2 votes)

2

Good afternoon. I have a question regarding Python. I have an if where has the conditional and else, the else it processes more than one file and I need to save all the information it reads inside a DataFrame, is there a way to do this?

The code I'm using:

for idx, arquivo in enumerate(fileLista):
    if arquivo == 'nome_do_arquivo_para_tratamento':
        df1 = pd.read_excel(arquivo, sheet_name = sheetName[idx], skiprows=1)
        df1.columns = df1.columns.str.strip()
        tratativaUm = df1[[informacoes das colunas que vão ser utilizadas]]

     else:
        df2 = pd.read_excel(arquivo, sheet_name = sheetName[idx], skiprows=1)
        df2.columns  = df2.columns.str.strip()
        TratativaDois = df2[[informacoes das colunas que vão ser utilizadas]]

####atribuir resultado de cada arquivo recebido no else

frames = [tratativaUm, tratativaDois] 
titEmpresa = pd.concat(frames)

Can anyone help me? Thank you

if python data.frame excel pandas

asked by anonymous 22.05.2018 / 20:43

1 answer

Doubt with SELECT and SUB SELECT in MySQL HashTable: Segmentation fault when inserting - C

score 2 · Accepted Answer

Hello

Assuming all read files have the same structure, each DataFrame created in else will have the same columns. So, just add the DataFrames read in the DataFrame result in each iteration.

# Inicializar DataFrame resultado
tratativaDois = pd.DataFrame()    

for idx, arquivo in enumerate(fileLista):
    if arquivo == 'nome_do_arquivo_para_tratamento':
        df1 = pd.read_excel(arquivo, sheet_name = sheetName[idx], skiprows=1)
        df1.columns = df1.columns.str.strip()
        tratativaUm = df1[[informacoes das colunas que vão ser utilizadas]]

    else:
       df2 = pd.read_excel(arquivo, sheet_name = sheetName[idx], skiprows=1)
       df2.columns  = df2.columns.str.strip()

       # Acrescentar novo DataFrame lido
       tratativaDois = tratativaDois.append(df2[[informacoes das colunas que vão ser utilizadas]])