在不使列表全局化的情况下访问 Pandas Dataframe 应用函数中的列表 [重复]

Question

我有一个 Pandas 数据框。对于这个框架中的每一行，我想进行一定的检查。如果检查产生True，那么我想将数据框行的某些列添加到列表结构中。

如何从 apply 函数中访问列表而无需创建全局列表变量？这可能吗？或者有更好的方法吗？

代码如下：

df.apply(checkFunction, axis=1)

checkFunction(row): 
    if (check == True):
        myList.append(row)
    return row

Answer 1

df.apply() 允许您传递位置参数和关键字。

def checkFunction(row, lst): 
    if (check == True):
        lst.append(row)
    return row

my_list = []
df.apply(checkFunction, axis=1, args=(my_list,))

不要称它为list，这与保留的list 关键字冲突。

注意：传递给args 的值必须是一个元组。我刚刚编辑了示例代码来展示这一点。