返回 DataFrame - Python

Question

早上好。

我有一个关于 Python 的问题。我有一个 if where 有条件和 else，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)

请注意，您可以创建一个 df 而不是 df1 和 df2，因为它与 read_excel 相同，然后取决于 folder 是否正确的，对 df

执行不同的操作

返回 DataFrame - Python

Returning in a DataFrame - Python

python

directory

excel

conditional

pandas