ndarray的axis问题

始终记不住np中axis是对应到哪个，还没系统地去学习下

先暂记两个常用的结果

1.[:,np.newaxis] 与 [np.newaxis, :]

注：这是ndarray才有的分片方法(np重写了[]操作符吧大概)，list不能这么干

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

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)

 

转载于:https://www.cnblogs.com/peanutk/p/9943294.html