始終記不住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)