gpt4 book ai didi

python - 分组和平均 NumPy 矩阵

转载 作者:太空狗 更新时间:2023-10-29 18:21:07 27 4
gpt4 key购买 nike

假设我有一个像这样的任意 numpy 矩阵:

arr = [[  6.0   12.0   1.0]
[ 7.0 9.0 1.0]
[ 8.0 7.0 1.0]
[ 4.0 3.0 2.0]
[ 6.0 1.0 2.0]
[ 2.0 5.0 2.0]
[ 9.0 4.0 3.0]
[ 2.0 1.0 4.0]
[ 8.0 4.0 4.0]
[ 3.0 5.0 4.0]]

什么是对按第三列编号分组的行进行平均的有效方法?

预期的输出是:

result = [[  7.0  9.33  1.0]
[ 4.0 3.0 2.0]
[ 9.0 4.0 3.0]
[ 4.33 3.33 4.0]]

最佳答案

一个紧凑的解决方案是使用 numpy_indexed (免责声明:我是它的作者),它实现了一个完全矢量化的解决方案:

import numpy_indexed as npi
npi.group_by(arr[:, 2]).mean(arr)

关于python - 分组和平均 NumPy 矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29291279/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com