2

我想维护一个字符串列表，并继续检查列表中是否存在不同的变量。

我正在做的是：

li=[one,two,three]
for x in different_x_values:
  if(x in li):
    print 'yes'

有没有比使用x in li它检查某个字符串是否在其他字符串集合中更快的方法？

4

4 回答 4

4

检查更新的python版本中的内置“set（）”数据类型 - 我认为它应该比列表更快。字典键曾经是快速的方法，但我认为现在集合相等或更快，所以如果你使用的是 2.5 之前的 python 版本，你可以执行以下操作（不管值是什么字典是）：

li = {'one':1, 'two': 1, 'three':1}
for x in different_values:
    if x in li:
        print 'yes'

更新版本，只需将 li 设为 set()。

于 2012-11-02T05:09:44.310 回答
4

为此使用一个set对象。

Python2> s = set(["one", "two", "three"])
Python2> "one" in s
True

Python2> s2 = set(["three", "four", "five"])

路口：

Python2> s & s2
set(['three'])

Python2> x = "three"
Python2> x in s & s2
True

我认为这就是你想要的，两组的交集。我认为，如果您了解集合对象，那么意图也更具可读性。

于 2012-11-02T05:09:55.550 回答
4 
print set(different_values).intersection(li)

应该工作相同......可能会更快..

>>> other_values = ["franks","pies","pie","beans"]
>>> li = ["apple","pie","franks","and","beans"]
>>> set(other_values).intersection(li)
set(['franks', 'beans', 'pie'])

这里有一些基准

In [1]: other_values = range(1000)
In [2]: li = range(500,1000)
In [3]: %timeit set(other_values).intersection(li)
10000 loops, best of 3: 78.6 us per loop
In [4]: %timeit [x for x in other_values if x in li]
100 loops, best of 3: 8.7 ms per loop
In [5]: %timeit set(other_values) & set(li)
10000 loops, best of 3: 97.6 us per loop

所以看起来设置交叉点要快得多

于 2012-11-02T05:10:59.930 回答
1

我相信您已经得到了答案，这是为了完整起见。出于好奇，我对提到的方法进行了测试，得到了以下结果。

size=主列表的大小
setopttime: 使用s1&s2的那个 setintersectiontime:
使用交集操作的
那个 bruteforcetime:burte force time

大小=1000
setoptime :0:00:00.001000 setintersectiontime
:0:00:00.001000
bruteforcetime:0:00:00.005000

大小=10000
setoptime :0:00:00.001000 setintersectiontime
:0:00:00.010000
bruteforcetime:0:00:00.367000

大小=100000
setoptime :0:00:00.001000 setintersectiontime
:0:00:00.115000
bruteforcetime:0:00:35.444000

方法 1 (setopttime) 胜出。如果您想检查，这是代码。

import timeit
import datetime
def getList(size):
    s=[]
    for i in range(size):
        s.append(str(i))
    return s

def getSubList(size):
    s=[]
    for i in range(size/2):
        s.append(str(i))
    return s

def testSet(size):
    s=set(getList(size))
    sublist=set(getSubList(size))
    list(s&sublist)


def testIntersection(size):
    s=set(getList(size))
    sublist=set(getSubList(size))
    list(s.intersection(sublist))

def bruteForce(size):
    s=getList(size)
    sublist=getSubList(size)
    final=[]
    for elem in sublist:
        try:
            if s.index(elem)!=-1:
                final.append(elem)
        except ValueError:
            pass
currentsize=1000
for i in range(3):
    t1=datetime.datetime.now()
    testSet(1000)
    t2=datetime.datetime.now()
    testIntersection(currentsize)
    t3=datetime.datetime.now()
    bruteForce(currentsize)
    t4=datetime.datetime.now()
    print "Size="+str(currentsize)
    print "setoptime:"+str(t2-t1)
    print "setintersectiontime:"+str(t3-t2)
    print "bruteforcetime:"+str(t4-t3)
    currentsize=currentsize*10
于 2012-11-02T05:09:40.677 回答