如何在数据帧上使用 def-return 或 for-in 语句来避免 python /pandas 中的代码重复

Question

有人可以看看下面的代码并建议我做错了什么。

我有 2 个熊猫数据框 - df 和 x1 两者具有相同的列和列名

我必须为 df.Date_Appointment、x1.Date_Appointment 和 df.Date_Scheduled 和 x1.Date_Scheduled 执行以下代码集。因此，为列和数据框创建了一个列表。

我正在尝试编写单个代码，但显然我做错了什么。请指教。

import pandas as pd
df = pd.read_csv(file1.csv)
x1 = pd.read_csv(file2.csv)

# x1 is a dataframe created after filtering on one column. # df and x1 have same number of columns and column names # x1 is a subset of df``

dataframe = ['df','x1']
column = ['Date_Appointment', 'Date_Scheduled']

def df_det (dataframe.column):

    (for df_det in dataframe.column :

        d_da = df_det.describe()
        mean_da = df_det.value_counts().mean()
        median_da = df_det.value_counts().median()
        mode_da = df_det.value_counts().mode()

        print('Details of all  appointments', '\n', 

         d_da, '\n',
         'Mean = ', mean_da,'\n', 
         'Median = ', median_da,'\n',
         'Mode = ',mode_da,'\n'))

请注明步骤。 提前谢谢你。

Answer 1

看起来你的函数应该有两个参数——dataframe 和column——它们都是列表，所以我将名称设为复数。

然后你需要遍历每个参数。请注意，您还在函数中分配了一个与您的函数同名的数据框，因此我更改了函数的名称。

dataframes = [dataframe1, dataframe2]
columns = ['Date_Appointment', 'Date_Scheduled']

def summary_stats(dataframes, columns):
    for df in dataframes:
        for col in cols:
            df_det = df.loc[:, col]
            # print summary stats about df_det