Hướng dẫn get index of max value in numpy array python - lấy chỉ mục của giá trị tối đa trong python mảng numpy

Trả về các chỉ số của các giá trị tối đa dọc theo một trục.

Mảng đầu vào.

Theo mặc định, chỉ mục được vào mảng phẳng, nếu không dọc theo trục được chỉ định.

Nếu được cung cấp, kết quả sẽ được chèn vào mảng này. Nó nên có hình dạng và dtype thích hợp.

Nếu điều này được đặt thành TRUE, các trục bị giảm được để lại trong kết quả là kích thước với kích thước một. Với tùy chọn này, kết quả sẽ phát sóng chính xác so với mảng.

Mới trong phiên bản 1.22.0.

Mảng các chỉ số vào mảng. Nó có hình dạng giống như A.Shape với kích thước dọc theo trục được loại bỏ. Nếu Keepdims được đặt thành True, thì kích thước của trục sẽ là 1 với mảng kết quả có hình dạng giống như A.Shape.

Ghi chú

Trong trường hợp nhiều lần xuất hiện của các giá trị tối đa, các chỉ số tương ứng với lần xuất hiện đầu tiên được trả về.

Ví dụ

>>> a = np.arange(6).reshape(2,3) + 10
>>> a
array([[10, 11, 12],
       [13, 14, 15]])
>>> np.argmax(a)
5
>>> np.argmax(a, axis=0)
array([1, 1, 1])
>>> np.argmax(a, axis=1)
array([2, 2])

Các chỉ mục của các phần tử tối đa của mảng N chiều:

>>> ind = np.unravel_index(np.argmax(a, axis=None), a.shape)
>>> ind
(1, 2)
>>> a[ind]
15

>>> b = np.arange(6)
>>> b[1] = 5
>>> b
array([0, 5, 2, 3, 4, 5])
>>> np.argmax(b)  # Only the first occurrence is returned.
1

>>> x = np.array([[4,2,3], [1,0,3]])
>>> index_array = np.argmax(x, axis=-1)
>>> # Same as np.amax(x, axis=-1, keepdims=True)
>>> np.take_along_axis(x, np.expand_dims(index_array, axis=-1), axis=-1)
array([[4],
       [3]])
>>> # Same as np.amax(x, axis=-1)
>>> np.take_along_axis(x, np.expand_dims(index_array, axis=-1), axis=-1).squeeze(axis=-1)
array([4, 3])

Thiết lập giữ thành đúng,

>>> x = np.arange(24).reshape((2, 3, 4))
>>> res = np.argmax(x, axis=1, keepdims=True)
>>> res.shape
(2, 1, 4)