8
Có cách nào để lấy max và argmax bằng một cú đánh không?Numpy argmax. Làm thế nào để tính toán cả max và argmax?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
Có cách nào nhanh nhất để làm điều đó, với một cái gì đó như:
[maximum,index]=function(a)
Tốt điểm trên tiết kiệm thời gian tính toán, tuy nhiên, nếu cuộc gọi argmax ban đầu là sắp xếp một mảng 2D trên trục thứ hai, thì câu lệnh thứ hai có thể sẽ xảy ra lỗi. – PaulDong
Bạn có thể khôi phục bằng cách sử dụng unravel_index (https://docs.scipy.org/doc/numpy/reference/generated/numpy.unravel_index.html) – PaxRomana99