从 function 返回 dataframe

Question

I'm having a little issue with the following code:

def process_row(row):
    liens = row['Liens']
    df_excel = pd.read_excel(liens)
    df_excel['Name'] = row['Name']
    print(df_excel)
    return df_excel

result_df = new_df.apply(process_row, axis=1)

new_df is a dataframe where 'Liens' is a column with multiple windows link to excel such as C:/Documents/.../test.xlsx When printing df_excel it does display the data in the excel file correctly, but not in result_df, it returns me a serie that is different from excel_df and I have no idea why... I would like to have a dataframe result_df with all the data from the multiple excel file regrouped.

Thanks for your help!!

Answer 1

If your goal is to read a bunch of excel files based on information in a dataframe and combine the resulting dataframes, something like this should do the trick:

import pandas as pd


def main():

    new_df = pd.DataFrame({"col_a": ["a", "b", "c"]})

    def process_row(row) -> pd.DataFrame:

        other_df = pd.DataFrame({"col_b": ["d", "e", "f"]})
        other_df["col_a"] = row.col_a

        return other_df

    # Iterate over all rows, process the row and concat the resulting dataframes
    result_df = pd.concat(
        map(process_row, new_df.itertuples()),
        ignore_index=True
    )

    print(result_df)


if __name__ == "__main__":
    main()

The output is as follows:

  col_b col_a
0     d     a
1     e     a
2     f     a
3     d     b
4     e     b
5     f     b
6     d     c
7     e     c
8     f     c

You can substitute your process_row and dataframe here.