numpy中的axis的理解

2021-06-21  本文已影响0人  术枚派

在使用numpy的时候,我们需要操作矩阵,这个时候就需要指定axis.

axis的意思是形容动作的一个副词,其中axis=0是指跨行操作,axis=1是跨列操作。


举个例子,numpy.argmax是对矩阵由大到小进行排序,返回各自的ID。

那么np.argmax(data, axis=1)

表示对数据跨列进行排序,返回的是每一行最大的ID到最小的ID。

上一篇下一篇

猜你喜欢

热点阅读