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) 。