什么是一种优雅的方式来返回没有循环的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]