将函数应用于dataframe列？

Question

我有以下函数（一个以列作为输入的热编码函数）。我基本上想把它应用到我的数据框中的一列，但似乎不明白出了什么问题。

def dummies(dataframe, col):
    dataframe[col] = pd.Categorical(dataframe[col])
    pd.concat([dataframe,pd.get_dummies(dataframe[col],prefix = 'c')],axis=1)

df1 = df['X'].apply(dummies)

猜我怎么称呼它有问题？

Answer 1

如果您想将其应用于该列，则不需要创建新的数据帧。这是正确的语法。请阅读文件。

df['X'] = df['X'].apply(lambda x : dummies(x))

Answer 2

您需要确保您正在从函数返回一个值，而当前您没有。。此外，当您将函数应用于列时，基本上是将列中每一行的值传递给函数，因此您的函数设置错误。。通常您会这样做：

def function1(value):
    new_value = value*2 #some operation
    return new_value

然后：

df['X'].apply(function1)

当前，您的函数设置为获取整个df和列的名称，因此，如果您这样调用函数，则函数可能会工作：

df1 = dummies(df, 'X')

但是你还是需要添加一个返回语句