numpy.argmax(axis)
给出这一维度上最大值的索引,且这一维消失
一维
import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
b = np.argmax(a)
print(b)
print(b.shape)
print(type(b))
输出
4
()
<class 'numpy.int64'>
二维
axis=0
import numpy as np
a = np.array([[1, 5, 5, 2],
[9, 6, 2, 8],
[3, 7, 9, 1]])
b = np.argmax(a, axis=0)
print(b)
print(b.shape)
print(type(b))
输出
[1 2 2 1]
(4,)
<class 'numpy.ndarray'>
axis=1
b = np.argmax(a, axis=1)
print(b)
print(b.shape)
输出
[1 0 2]
(3,)
三维
axis=0
import numpy as np
a = np.array([
[
[1, 5, 5, 2],
[9, -6, 2, 8],
[-3, 7, -9, 1]
],
[
[-1, 5, -5, 2],
[9, 6, 2, 8],
[3, 7, 9, 1]
]
])
b = np.argmax(a, axis=0)
print(b)
print(b.shape)
print(type(b))
输出
[[0 0 0 0]
[0 1 0 0]
[1 0 1 0]]
(3, 4)
<class 'numpy.ndarray'>
axis=1
b = np.argmax(a, axis=1)
print(b)
print(b.shape)
输出
[[1 2 0 1]
[1 2 2 1]]
(2, 4)
axis=2
输出
[[1 0 1]
[1 0 2]]
(2, 3)
参考自:
https://www.cnblogs.com/zhouyang209117/p/6512302.html