计算Nympy 2D数组的mean（），该数组按单独列表中的值分组，该列表中的字符串对应于2D数组中的每一行

Question

我正在参加一个关于Python数据分析的课程（Numpy、Pandas等）

我们有一个赋值，我们应该根据另一个列表的值计算数组的mean（）。 这似乎有点不清楚，因此下面是一个示例：

list = ['A','A','A','A','B','B','B','B']
array = [ [5.1, 3.5, 1.4, 0.2],
        [4.9, 3. , 1.4, 0.2],
        [4.7, 3.2, 1.3, 0.2],
        [4.6, 3.1, 1.5, 0.2],
        [5. , 3.6, 1.4, 0.2],
        [5.4, 3.9, 1.7, 0.4],
        [4.6, 3.4, 1.4, 0.3],
        [5. , 3.4, 1.5, 0.2] ]

列表值对应于数组中行的类别，要求我们计算按A和B分组的每列的平均值。 我想这可以通过将数据转换成Pandas数据帧来实现——但是这个赋值属于Numpy，所以我想我们应该在没有Pandas的情况下解决它

我苦苦挣扎，谷歌搜索也无济于事。 非常感谢您的帮助

谢谢

B.R。 安德斯

Answer 1

这能解决你的问题吗

import numpy as np

list = ['A','A','A','A','B','B','B','B']
array = np.array([ [5.1, 3.5, 1.4, 0.2],
        [4.9, 3. , 1.4, 0.2],
        [4.7, 3.2, 1.3, 0.2],
        [4.6, 3.1, 1.5, 0.2],
        [5. , 3.6, 1.4, 0.2],
        [5.4, 3.9, 1.7, 0.4],
        [4.6, 3.4, 1.4, 0.3],
        [5. , 3.4, 1.5, 0.2] ])
        
idxA = np.array([i for i, x in enumerate(list) if x == "A"])
idxB = np.array([i for i, x in enumerate(list) if x == "B"])

print(np.mean(array[idxA]))
print(np.mean(array[idxB]))

输出：

2.40625
2.5875000000000004