Trả về các chỉ số của các giá trị tối đa dọc theo một trục.
Parametersaarray_likeaarray_likeMảng đầu vào.
Trục, tùy chọnint, optionalTheo 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.
OutArray, tùy chọnarray, optionalNế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.
ReturnSindex_arrayndarray của intsindex_arrayndarray of intsMả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]