अजगर/Numpy - लपेटें स्लाइस सरणी

Question

के अंत के आसपास मैं दो -1 डी सरणियों है, एक ब्याज (क) में से कुछ मानों और दूसरा जो कि सरणी में सूचकांक प्रदान करता है (ख)। मुझे पता है कि बी में मान हमेशा बढ़ते हैं, एक बिंदु को छोड़कर (कहीं भी हो सकता है) जहां संख्या कम हो जाती है क्योंकि यह अंत से सरणी की शुरुआत तक रोल हो जाती है। नीचे दी गई विधि काम करने लगती है, लेकिन मुझे लगता है कि एक क्लीनर तरीका मौजूद होना चाहिए। क्या कोई बेहतर कुछ सुझाव दे सकता है? धन्यवाद।

कोड:

import numpy as np 
a = np.arange(12) 
b = np.array([5, 9, 2, 4]) 
#I want to generate these: 
#[5,6,7,8,9] 
#[9,10,11,0,1,2] 
#[2,3,4] 
#[4,5] 

a = np.roll(a, -b[0], axis=0) 
# Subtract off b[0] but ensure that all values are positive 
b = (b-b[0]+len(a))%len(a) 
for i, ind in enumerate(b): 
    if i < len(b)-1: 
     print a[b[i]:b[i+1]+1] 
    else: 
     print np.hstack((a[b[i]:len(a)], a[0])) 

Answer 1

थोड़ा कम

, लेकिन शायद मैं अभी भी बेहतर कर सकते हैं ...

import numpy as np 

a = np.arange(12) 
b = np.array([5, 9, 2, 4]) 
b = np.append(b, b[0]) 

for i in range(0, len(b)-1): 
    print np.roll(a, len(a)-b[i])[:b[i+1]-b[i]+1] 

Answer 2

सुनिश्चित नहीं हैं कि अगर यह मदद करता है, लेकिन की स्मृति के साथ खिलवाड़ के बिना एक तेजी से रास्ता

import numpy as np 

a = np.arange(12) 
b = np.array([5, 9, 2, 4]) 
b = np.append(b, b[0]) 

b2 = b.copy() 

b2[(np.diff(b)<0).nonzero()[0]+1:] += a.size 

print [np.take(a, np.r_[b2[i]:b2[i+1]+1], mode='wrap') for i in range(b.size-1)] 

print [np.roll(a, len(a)-b[i])[:b[i+1]-b[i]+1] for i in range(b.size-1)] 

%timeit [np.take(a, np.r_[b2[i]:b2[i+1]+1], mode='wrap') for i in range(b.size-1)] 
# 10000 loops, best of 3: 28.6 µs per loop 

%timeit [np.roll(a, len(a)-b[i])[:b[i+1]-b[i]+1] for i in range(b.size-1)] 
# 10000 loops, best of 3: 77.7 µs per loop