Numpy: argsort बहुआयामी सरणियों के लिए काम नहीं करता है

argsort() ठीक काम कर रहा है, लेकिन अनुक्रमण उस तरह से काम नहीं करता है जिसकी आप अपेक्षा कर रहे हैं।

http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#advanced -indexing

ये कोशिश करें:

a = np.random.random([5,5])
i = np.arange(len(a))[:, np.newaxis]
j = np.argsort(a, axis=1)
a_sorted = a[i, j]
np.sort(a, axis=1)

मैं उसी तरह उलझा हुआ था। दस्तावेज़ीकरण विशेष रूप से बताता है,

रिटर्न
index_array : ndarray, int
इंडेक्स की सरणी जो निर्दिष्ट अक्ष के साथ a सॉर्ट करती है।
दूसरे शब्दों में, a[index_array] एक क्रमबद्ध a उत्पन्न करता है।

जो केवल 1D सरणी पर लागू होता है। क्या दस्तावेज को यह निर्दिष्ट करने के लिए बढ़ाया जाना चाहिए कि यह एनडी सरणी के मामले में नहीं है?

@lzkelley हां, दस्तावेज़ीकरण में सुधार के लिए एक पैच का बहुत स्वागत होगा।

@ शॉयर कूल, मैं इसके लिए एक पीआर

यह अच्छा होगा यदि एनडी सरणी को सॉर्ट करने के लिए वास्तव में आउटपुट का उपयोग करने का एक उदाहरण था।

यह भी अच्छा होगा यदि a[np.arange(np.shape(a)[0])[:,np.newaxis], np.argsort(a)] से अधिक पठनीय समाधान हो

हो सकता है कि @seberg इसे अपने अनुक्रमण फ़ंक्शन परिवर्धन में से एक बना सके।

मुझे नहीं लगता कि यह बहुत स्पष्ट रूप से अनुक्रमण को फिट करता है, एक पिक फ़ंक्शन या तो शायद एक आसान मैच है।

एक नए समारोह के लिए +1।

क्या argsort (जैसा कि sort ) का संपूर्ण उद्देश्य अनुक्रमण के लिए उपयोग नहीं किया जाना है?

argsort एक अक्ष के साथ अनुक्रमित करने के लिए अनुक्रमित की आपूर्ति करता है, लेकिन वास्तव में अनुक्रमणिका का उपयोग करके उस अक्ष के साथ क्रमबद्ध करना अनावश्यक रूप से क्रियात्मक लगता है। जब तक कोई आसान तरीका न हो, मुझे पता नहीं है।

https://github.com/numpy/numpy/blob/5becd12914d0402967205579d6f59a98151e0d98/doc/neps/indexing.rst#संबंधित - प्रश्न

एक पिक फंक्शन या तो शायद एक आसान मैच है।

यह #8708 . है