什麼是一種優雅的方式來返回沒有循環的python中的列表的索引？

Question

說我有一個清單t = list(i for i in np.arange(0.5,10.5,0.5))。我想在t之內找到s=[3,5,7]的指數頭寸。我知道我可以通過t.index(s)循環來做到這一點，但有沒有更優雅的方式？

Answer 1

認識到你總是需要一些循環很重要。即使t.index做了一個循環，它只是隱藏它從你！不過據我所知，Python標準庫中沒有函數可以避免s的顯式循環。

---

但是一個可以使它更有效的（我的意思是比[t.index(needle) for needle in s]更有效）！特別是如果你的清單是排序的。

您已使用NumPy所以np.searchsorted肯定是快：

import numpy as np 

t = np.arange(0.5,10.5,0.5) 
s = [3, 5, 7] 

np.searchsorted(t, s) 

如果你想留在Python裏面還有bisect模塊，至少可以節省一些（隱含的）循環的：

from bisect import bisect_left 

def index(a, x): # Taken from the bisect documentation 
    'Locate the leftmost value exactly equal to x' 
    i = bisect_left(a, x) 
    if i != len(a) and a[i] == x: 
     return i 
    raise ValueError 

indices = [index(t, needle) for needle in s]