द्वितीयक सरणी सूचकांकों के आधार पर तत्वों को निकालने के लिए पूर्णांक सरणी अनुक्रमण का उपयोग करना

दिए गए परिदृश्य में, लक्ष्य एक से विशिष्ट तत्वों को पुनः प्राप्त करना है सरणी ए दूसरे सरणी बी में निर्दिष्ट सूचकांकों का उपयोग कर रही है। np.take या np.choose पर भरोसा करने के बजाय, एक अधिक सरल तरीका नियोजित करना है NumPy की पूर्णांक सरणी अनुक्रमणिका:

A[np.arange(A.shape[0]),B.ravel()]

यहां बताया गया है कि यह कोड वांछित परिणाम कैसे प्राप्त करता है:

• np.arange(A.shape[0]) 0 से (A.shape[0] तक सूचकांकों की एक सरणी बनाता है - 1), ए की पंक्तियों का प्रतिनिधित्व करता है। इंडेक्सिंग ऑपरेशन ए[...] बी से पंक्ति सूचकांकों और कॉलम सूचकांकों का उपयोग करके ए से तत्वों को पुनर्प्राप्त करता है।
• यह दृष्टिकोण विशेष रूप से तब उपयोगी होता है जब बी एक 1डी सरणी या एक सूची है स्तंभ सूचकांकों का. फ़्लैटनिंग ऑपरेशन को छोड़ देने से, कोड और भी सरल हो जाता है:
• A[np.arange(A.shape[0]),B]

उदाहरण:

A[np.arange(A.shape[0]),B]