我有一个熊猫数据帧。对于此框架中的每一行,我想进行一定的检查。如果检查产生 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
的值必须是元组。我刚刚编辑了示例代码来显示这一点。