创建一个函数来在列上分解数据框

Question

我已经编写了一些代码来在列上分解 df 。

df=df.assign(col=df['col'].str.split(',')).explode('col')

我现在想让一个函数执行相同的操作：

def explode(df,col_ind,col:str):
df=df.assign(col_ind=df[col].str.split(',')).explode(col)
return df

explode(df5,beans,'beans')

我确实收到以下错误：

名称错误：名称“beans”未定义

块引用

有人知道如何改进该功能以使其正常工作吗？

Answer 1

您调用未定义的 bean 变量。要从字符串定义 kwarg，您必须使用 ** 将其作为 kwarg 字典传递：

def explode(df, col_ind, col:str):
    df=df.assign(**{col_ind : df[col].str.split(',')}).explode(col)
    return df

explode(df5, 'beans', 'beans')