python - 在使用点表示法的函数中包含列名作为参数

Question

我想将列名作为此函数可接受的参数包含在内，以使其更加灵活。

def func(df, column):
    return df.column.str.split('', n = 5, expand=True)

正如预期的那样，func(my_df, columnX) 返回 NameError: name 'columnX' is not Define。

有解决办法吗？

Answer 1

您可以像访问字典字段一样访问您的列。另一个建议是不要将对象命名为类名(将 DataFrame 替换为 df)。最后需要返回获得的对象。

def func(df, column):
    return df[column].str.split('', n = 5, expand=True)