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

Question

我有以下值：

grade_list = [[99 73 97 98] [98 71 70 99]]

excercise_list = ['1' '2']

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

输出应如下所示：[[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 数组中并将类型设置为 float（您的练习是字符串）：result = np.array([excercise_list, mean_grades]).astype(float).