将功能应用于数据框列？

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

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

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

然后：

df['X'].apply(function1)

当前，您的函数设置为采用整个df和一列的名称，因此，如果您按以下方式调用它，则函数可能会起作用：

df1 = dummies(df, 'X')

但是您仍然需要添加一个return语句

Answer 2

如果要将其应用于该列，则无需创建新的数据框。 这是正确的语法。 请阅读文档 。

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