Function 丢弃数据帧中的无效值，但随后返回具有无效值的原始数据帧

Question

我创建了这个简单的 function：

def cleanup_data(*argv):
    for df in argv:
        df = df.dropna()
    return argv

但是，如果我调用 cleanup_data(df1, df2)，稍后我会这样做：

df1.isnull().values.any()

或者

df2.isnull().values.any()

我明白了。

我的代码有什么问题？

Answer 1

您不会返回更新的数据帧，而是返回未更改的argv 。 以下是如何使用列表推导式返回更新后的数据框列表：

def cleanup_data(*argv):
    return [df.dropna() for df in argv]

或者，您可以使df.dropna在数据帧上就地运行：

def cleanup_data(*argv):
    for df in argv:
        df.dropna(inplace=True)