python - 将函数应用于我的数据框中的所有列

Question

我无法将我的自定义函数应用到 dataframe:

df = pd.DataFrame({
        "a": ["related-1", "related-0", "related-1"],
        "b": ["request-1", "request-0", "request-1"],
        "c": ["offer-1", "offer-0", "offer-1"],
    })

def clean_column(df, column):
    return df[column].apply(lambda x: re.sub(".*-", "", x)).astype("int64")

clean_column(df, "a") # This works

我想将此函数应用于多个列:

df.applymap(lambda x: clean_column(df, x))

错误:KeyError:'related-1'

现在确定我在这里做错了什么，有人能告诉我这不起作用的原因吗？

Answer 1

apply 将整个列作为输入，而不仅仅是名称。所以你想要:

def clean_column(column):
    return column.apply(lambda x: re.sub(".*-", "", x)).astype("int64")

df.apply(clean_column)

但是，在您的情况下，您需要为每一列再次apply。换句话说，您想将函数应用于所有单元格。即applymap:

df.applymap(lambda x: re.sub(".*-", "", x)).astype("int64")

输出:

   a  b  c
0  1  1  1
1  0  0  0
2  1  1  1