एक Numpy सरणी

Question

के भीतर कई मान खोजें I इंडेक्स को खोजने के लिए एक numpy फ़ंक्शन की तलाश में है जिसमें वेक्टर (xs) के भीतर कुछ मान पाए जाते हैं। मान किसी अन्य सरणी (ys) में दिए जाते हैं। लौटाए गए सूचकांक को ys के आदेश का पालन करना होगा।

कोड में, मैं नीचे सूचीबद्ध सूची द्वारा सूची समझ को प्रतिस्थापित करना चाहता हूं।

>> import numpy as np 
>> xs = np.asarray([45, 67, 32, 52, 94, 64, 21]) 
>> ys = np.asarray([67, 94]) 
>> ndx = np.asarray([np.nonzero(xs == y)[0][0] for y in ys]) # <---- This line 
>> print(ndx) 
[1 4] 

क्या कोई तेज़ तरीका है?

धन्यवाद

Answer 1

बड़ा सरणियों xs और ys के लिए, आप इस के लिए बुनियादी दृष्टिकोण को बदलने के लिए तेजी से बनने के लिए की आवश्यकता होगी।

xs.sort() 
ndx = numpy.searchsorted(xs, ys) 

तो यह महत्वपूर्ण है xs के मूल आदेश रखने के लिए, तो आप इस दृष्टिकोण का उपयोग कर सकें, लेकिन आप याद करने की जरूरत है: आप xs छँटाई के साथ ठीक है, तो एक आसान विकल्प numpy.searchsorted() उपयोग करने के लिए है मूल सूचकांक:

orig_indices = xs.argsort() 
ndx = orig_indices[numpy.searchsorted(xs[orig_indices], ys)]