2014-11-04 13 views

Trả lời

9

Có lẽ một cái gì đó như thế này là nhanh ...

index=np.argmax(a) 
max=a(index) 
+0

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

+0

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

Các vấn đề liên quan