如何迭代pandas中的列标题

Question

我尝试遍历列标题并填充一个包含每个类别的平均收入的列表。 我遇到了这个错误，我尝试了很多不同的方法来修复它。 所以我的数据框如下所示（看起来我没有足够的点来粘贴图像，但这里有一个链接）： each row in the dataframe represent an idividual movie. column revenue_adj populated with revenue values, one column for each genre populated with TRUE/FALSE indicating that the movie falls under that specific genre.

这段代码工作得很好，但我希望在for循环中而不是在steadgenrev1中执行

genrev1= df.query('action == True')['revenue_adj'].mean()
genrev2 = df.query('adventure == True')['revenue_adj'].mean()
genrev3 = df.query('animation == True')['revenue_adj'].mean()
genrev4 = df.query('comedy == True')['revenue_adj'].mean()
genrev5 = df.query('crime == True')['revenue_adj'].mean()
genrev6 = df.query('documentary == True')['revenue_adj'].mean()
genrev7 = df.query('drama == True')['revenue_adj'].mean()
genrev8 = df.query('family == True')['revenue_adj'].mean()

我尝试了以下循环，但没有成功：

^{pr2}$

我得到一个错误说“gen is not defined” 而不是gen，我尝试过：

df[gen] == True
'@df[gen] == True'
'@gen == True'

我已经坚持了好几个星期了，如果有任何帮助，我将不胜感激！！在

Answer 1

你可以试试这样的方法：

df = pd.DataFrame({'Revenue':np.arange(100,1001,100),
                   'action':np.random.choice([True, False],10),
                   'comedy':np.random.choice([True, False],10),
                   'drama':np.random.choice([True, False],10)})

df.iloc[:,1:].apply(lambda x: pd.Series(df.loc[x,'Revenue'])).mean()

输出：

^{pr2}$