पायथन के न्यूमपी

Question

में उच्चतम मूल्य का सूचकांक निर्धारित करें मैं प्रत्येक पंक्ति के उच्चतम अधिकतम मूल्य के सूचकांक के साथ एक सरणी उत्पन्न करना चाहता हूं।

a = np.array([ [1,2,3], [6,5,4], [0,1,0] ]) 
maxIndexArray = getMaxIndexOnEachRow(a) 
print maxIndexArray 

[[2], [0], [1]] 

वहाँ एक np.argmax समारोह है, लेकिन यह मैं क्या चाहते हो प्रतीत नहीं होता है ...

Answer 1

argmax() समारोह आप क्या चाहते हैं क्या करता है:

print a.argmax(axis=1) 
array([2, 0, 1])