Возврат в DataFrame - Python

Question

Доброе утро.

У меня вопрос по поводу Python.У меня есть if, где есть условное и else, иначе он рендерит более одного файла, и мне нужно сохранить всю информацию, которую он читает внутри DataFrame, есть ли способ сделать это?

Код, который яиспользуя:

for idx, folder in enumerate(fileLista):
    if folder == 'filename_for_treatment':
        df1 = pd.read_excel(folder, sheet_name = sheetName[idx], skiprows=1)
        df1.columns = df1.columns.str.strip()
        tratativaUm = df1[[column information to be used]]

     else:
        df2 = pd.read_excel(folder, sheet_name = sheetName[idx], skiprows=1)
        df2.columns  = df2.columns.str.strip()
        TratativaDois = df2[[column information to be use]]

####assign result of each file received in the else

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

Может кто-нибудь помочь мне, возможно ли это сделать?Спасибо

Answer 1

вы можете сделать это, добавив свои фреймы данных в список, например:

list_df_tratativaDois = []
for idx, folder in enumerate(fileLista):
    df = pd.read_excel(folder, sheet_name = sheetName[idx], skiprows=1)
    df.columns = df.columns.str.strip()
    if folder == 'filename_for_treatment':
        tratativaUm = df[[column information to be used]]
     else:
        list_df_tratativaDois.append(df[[column information to be use]])

titEmpresa = pd.concat([tratativaUm]+list_df_tratativaDois)

Обратите внимание, что вместо df1 и df2 вы можете просто создать df, как это было то же самоеread_excel и затем, в зависимости от того, является ли folder правильным, выполните другое действие на df