ndarray的axis問題

始終記不住np中axis是對應到哪一個,還沒系統地去學習下學習

先暫記兩個經常使用的結果spa

1.[:,np.newaxis] 與 [np.newaxis, :]code

注:這是ndarray纔有的分片方法(np重寫了[]操做符吧大概),list不能這麼幹blog

a = np.array([1,2,3,4]) a[:,np.newaxis] array([[1], [2], [3], [4]]) a[np.newaxis,:] array([[1, 2, 3, 4]])

 

2.argmax()class

a = np.arange(24).reshape(4,6) a.argmax() 23 a.argmax(axis=0) array([3, 3, 3, 3, 3, 3], dtype=int64) a.argmax(axis=1) array([5, 5, 5, 5], dtype=int64)
相關文章
相關標籤/搜索