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)