访问 Pandas 数据帧应用函数中的列表,而无需使列表全局



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

如何在不创建全局列表变量的情况下从 apply 函数中访问列表?这可能吗?或者有更好的方法可以做到这一点吗?

代码如下所示:

df.apply(checkFunction, axis=1)
checkFunction(row): 
    if (check == True):
        myList.append(row)
    return row

>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的值必须是元组。我刚刚编辑了示例代码来显示这一点。

最新更新