1. 程式人生 > >對運用axis的理解

對運用axis的理解

基本思路
⚫本例中以元素全部為1的2345的四維陣列為例
⚫案例的axis=n,是求第n維的和,即np.sum(陣列,axis=n)
⚫案例中的所有元素按照維度編號,例:三維陣列第1個:000,四
維陣列第2個:0002
⚫本例的思維結構為:原來是n維陣列,用axis=你,對第n維度求和而生成n-1維數
組。新生成的陣列形狀是在原來形狀基礎上刪除被求和的維度。比如2
345形狀的四維陣列,由於對第1維度求和則第1維度的3被刪除,變成245形狀三維陣列。
在這裡插入圖片描述