开发者社区> 问答> 正文

如何计算分组dataframe的平均值,并将分组列保留为另一列?

如何计算分组dataframe的平均值,并将分组列保留为另一列?

展开
收起
游客y244y7ln2rlpa 2021-12-05 20:39:19 733 0
1 条回答
写回答
取消 提交回答
  • df = pd.DataFrame({'fruit': ['apple', 'banana', 'orange'] * 2,
                       'rating': np.random.rand(6),
                       'price': np.random.randint(0, 15, 6)})
    
    # 按fruit分组后,price列的平均值,并将分组置为一列
    out = df.groupby('fruit', as_index=False)['price'].mean()
    print(out)
    
    #>	    fruit  price
    	0   apple    4.0
    	1  banana    6.5
    	2  orange   11.0
    33.如何获取两列值元素相等的位置(并非索引)
    df = pd.DataFrame({'fruit1': np.random.choice(['apple', 'orange', 'banana'], 3),
                        'fruit2': np.random.choice(['apple', 'orange', 'banana'], 3)})
    
    print(df)
    # 获取两列元素相等的行
    np.where(df.fruit1 == df.fruit2)
    
    #>	   fruit1  fruit2
    	0   apple  banana
    	1   apple   apple
    	2  orange   apple
    
    #>	(array([1], dtype=int64),)
    
    
    2021-12-05 22:23:08
    赞同 展开评论 打赏
问答地址:
问答排行榜
最热
最新

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载