Pandas iterrows 在函数内不起作用

Question

我正在创建一个小规模的调度脚本，但遇到了一些 iterrows 问题。这些是非常小的 df，所以时间很短（6 行，可能 7/8 列），虽然我猜这些循环不是最有效的 - 我对此很陌生！

这是我已经拥有的

data = {'Staff 1': ['9-5', '9-5', '9-5', '9-5', '9-5'], 
         'Staff 2': ['9-5', '9-5', '9-5', '9-5', '9-5'],
            'Staff 3': [ '9-5', '9-5', '9-5', '9-5', '9-5']}
dataframe_1 = pd.DataFrame.from_dict(data, orient='index',
                        columns=['9/2/19', '9/3/19', '9/4/19', '9/5/19', '9/6/19'])

data2 = {'Name': ['Staff 1','Staff 2','Staff 3'], 'Site': ['2','2','2'], 'OT':['yes','yes','no'],
         'Days off':['','9/4/19','9/4/19'], '':['','','9/5/19']}

dataframe_2 = pd.DataFrame.from_dict(data2)


def annual_leave(staff, df):
    df = df.reset_index(drop=True)
    for index, row in df.iterrows():
        days_off = []
        if df.loc[index,'Name'] == '{}'.format(staff):
            for cell in row:                
                days_off.append(cell)
            del days_off[0:3]
        else:
            pass

        return days_off

for index, row in dataframe_1.iterrows():
    print(annual_leave(index, dataframe_2))

我在一些地方添加了一些“打印（索引）”，看看我是否可以找出问题所在。

我发现底部的 iterrows 循环贯穿每一行。但是，函数中的 itterrows 循环只查看第一行，我不明白为什么。

我正在尝试检查 dataframe_1 中的每个员工姓名（索引），并根据 dataframe_2 中的列名检查该员工姓名。然后，我想删除 dataframe_2 中该特定行的前 3 列（因此是列表和 del days_off[0:3]）。

但在本例中，它为“Staff 1”、“Staff 2”和“Staff 3”运行底部 for iterrows 循环（函数外部）。但函数内的 itterrows 循环仅检查“Staff 1”名称。

这意味着它仅适用于“Staff 1”，但当为“Staff 2”调用该函数时，它仅在 dataframe_2 的第一行中检查“Staff 2” - 并没有找到它，因为它在第二行。

这有意义吗？

非常感谢任何帮助。

Answer 1

您是否尝试过调用该函数？执行您给出的上述代码只会创建一个函数，您需要通过传递适当的参数来调用它以查看输出。我在代码中找不到任何其他错误。如有错误请指正。

Answer 2

这是因为您的 return 在 iterrows for 循环内。 return 需要不缩进才能在 for 循环之外。