NumPy - 将两个 arrays 格式化为一个多维数组

Question

我有以下价值观：

grade_list = [[99 73 97 98] [98 71 70 99]]
excercise_list = ['1' '2']

使用 Numpy，我想将其转换为一个多维数组以获得每个练习的平均成绩（grade_list 中的第一项是指练习编号 1）

output 应如下所示： [[1. 2.] [91.75 84.5]] [[1. 2.] [91.75 84.5]]

这意味着平均。 练习#1 的成绩为 91.75，#2 的成绩为 84.5。

如何使用 numpy 进行转换？ 我已阅读有关NumPy 轴参数的信息，但不知道如何将它们放在一起。

Answer 1

轴 0 是第一个嵌套级别（两个列表），轴 1 是第二级（轴 0 中每个条目四个等级）。 您想要计算沿轴 1 的平均值，以便保留轴 0。 所以平均成绩是

mean_grades = np.mean(grade_list, axis=1) 。

然后将两个列表堆叠在另一个嵌套列表中，将其包装在 numpy 数组中并将类型设置为浮点（您的练习是字符串）：

result = np.array([excercise_list, mean_grades]).astype(float) 。