मेरे पास एक सरणी है, और मैं यह जानना चाहता हूं कि प्रत्येक पंक्ति के लिए अधिकतम मान कहां हैं। उसी पंक्ति में मेरे पास अधिकतम दो मान हो सकते हैं, इसलिए मैंने np.argwhere का उपयोग करने का निर्णय लिया।

np.argwhere(a == np.amax(a,axis=1))

लेकिन मेरी पहली पंक्ति के लिए, यह कुछ भी आउटपुट नहीं करता है और मुझे समझ में नहीं आता क्यों।

0
isaaccs 20 अक्टूबर 2020, 14:53

1 उत्तर

सबसे बढ़िया उत्तर

np.amax एक पंक्ति सरणी देता है, जबकि आप एक कॉलम चाहते हैं। इसलिए, आप गणना करने से पहले अपनी सरणी को दोबारा बदलना चाहते हैं:

np.argwhere(a == np.amax(a, axis=1).reshape((-1,1)))
1
Tristan Nemoz 20 अक्टूबर 2020, 12:02