AttributeError：“DataFrameGroupBy”对象没有属性“colnames”

Question

我有一个列名列表，如下所示：

colnames = ['split1', 'split2', 'split3', 'split4', 'split5', 'split6', 'split7']

我在一行代码中使用了它，我在其中迭代列表项，如下所示：

for i in range(0,x+2):
    df[d[i+1]] = df.groupby(d[i]).colnames[i].transform(lambda x : x.factorize()[0]+1)

这给了我属性错误：

AttributeError: 'DataFrameGroupBy' object has no attribute 'columns'

但如果我将 colname 项单独提供给代码，它就可以正常工作：

df['dot2'] = df.groupby('dot1').split1.transform(lambda x : x.factorize()[0]+1)

Answer 1

在df.groupby(d[i]).colnames[i] 中，colnames[i] 是一个变量。你可能需要做

df.groupby(d[i])[f'{colnames[i]}']