사전에 키 목록을 사용하여 해당 항목의 목록을 얻는 기본 제공 / 빠른 방법이 있습니까?
예를 들어
>>> mydict = {'one': 1, 'two': 2, 'three': 3}
>>> mykeys = ['three', 'one']mykeys사전에서 해당 값을 목록으로 가져 오려면 어떻게 해야합니까?
>>> mydict.WHAT_GOES_HERE(mykeys)
[3, 1]답변
목록 이해는 이것을 수행하는 좋은 방법 인 것 같습니다.
>>> [mydict[x] for x in mykeys]
[3, 1]답변
list-comp 이외의 다른 두 가지 방법 :
- 키를 찾을 수 없으면 빌드 목록 및 예외 발생 :
map(mydict.__getitem__, mykeys) Noneif 키가없는 빌드 목록 :map(mydict.get, mykeys)
또는를 사용 operator.itemgetter하면 튜플을 반환 할 수 있습니다.
from operator import itemgetter
myvalues = itemgetter(*mykeys)(mydict)
# use `list(...)` if list is required참고 : Python3에서는 map목록이 아닌 반복자를 반환합니다. list(map(...))목록에 사용하십시오 .
답변
약간의 속도 비교 :
Python 2.7.11 |Anaconda 2.4.1 (64-bit)| (default, Dec 7 2015, 14:10:42) [MSC v.1500 64 bit (AMD64)] on win32
In[1]: l = [0,1,2,3,2,3,1,2,0]
In[2]: m = {0:10, 1:11, 2:12, 3:13}
In[3]: %timeit [m[_] for _ in l] # list comprehension
1000000 loops, best of 3: 762 ns per loop
In[4]: %timeit map(lambda _: m[_], l) # using 'map'
1000000 loops, best of 3: 1.66 µs per loop
In[5]: %timeit list(m[_] for _ in l) # a generator expression passed to a list constructor.
1000000 loops, best of 3: 1.65 µs per loop
In[6]: %timeit map(m.__getitem__, l)
The slowest run took 4.01 times longer than the fastest. This could mean that an intermediate result is being cached
1000000 loops, best of 3: 853 ns per loop
In[7]: %timeit map(m.get, l)
1000000 loops, best of 3: 908 ns per loop
In[33]: from operator import itemgetter
In[34]: %timeit list(itemgetter(*l)(m))
The slowest run took 9.26 times longer than the fastest. This could mean that an intermediate result is being cached
1000000 loops, best of 3: 739 ns per loop따라서 목록 이해와 itemgetter가 가장 빠른 방법입니다.
업데이트 : 큰 무작위 목록과지도의 경우 약간 다른 결과가있었습니다.
Python 2.7.11 |Anaconda 2.4.1 (64-bit)| (default, Dec 7 2015, 14:10:42) [MSC v.1500 64 bit (AMD64)] on win32
In[2]: import numpy.random as nprnd
l = nprnd.randint(1000, size=10000)
m = dict([(_, nprnd.rand()) for _ in range(1000)])
from operator import itemgetter
import operator
f = operator.itemgetter(*l)
%timeit f(m)
%timeit list(itemgetter(*l)(m))
%timeit [m[_] for _ in l] # list comprehension
%timeit map(m.__getitem__, l)
%timeit list(m[_] for _ in l) # a generator expression passed to a list constructor.
%timeit map(m.get, l)
%timeit map(lambda _: m[_], l)
1000 loops, best of 3: 1.14 ms per loop
1000 loops, best of 3: 1.68 ms per loop
100 loops, best of 3: 2 ms per loop
100 loops, best of 3: 2.05 ms per loop
100 loops, best of 3: 2.19 ms per loop
100 loops, best of 3: 2.53 ms per loop
100 loops, best of 3: 2.9 ms per loop따라서이 경우 명확한 승자는 f = operator.itemgetter(*l); f(m)이고 외부인 은 분명 map(lambda _: m[_], l)합니다.
Python 3.6.4 업데이트 :
import numpy.random as nprnd
l = nprnd.randint(1000, size=10000)
m = dict([(_, nprnd.rand()) for _ in range(1000)])
from operator import itemgetter
import operator
f = operator.itemgetter(*l)
%timeit f(m)
%timeit list(itemgetter(*l)(m))
%timeit [m[_] for _ in l] # list comprehension
%timeit list(map(m.__getitem__, l))
%timeit list(m[_] for _ in l) # a generator expression passed to a list constructor.
%timeit list(map(m.get, l))
%timeit list(map(lambda _: m[_], l)
1.66 ms ± 74.2 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
2.1 ms ± 93.2 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
2.58 ms ± 88.8 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
2.36 ms ± 60.7 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
2.98 ms ± 142 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
2.7 ms ± 284 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
3.14 ms ± 62.6 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)따라서 Python 3.6.4의 결과는 거의 동일합니다.
답변
세 가지 방법이 있습니다.
모금 KeyError키가 발견되지 않는 경우 :
result = [mapping[k] for k in iterable]누락 된 키의 기본값입니다.
result = [mapping.get(k, default_value) for k in iterable]누락 된 키를 건너 뜁니다.
result = [mapping[k] for k in iterable if k in mapping]답변
이 시도:
mydict = {'one': 1, 'two': 2, 'three': 3}
mykeys = ['three', 'one','ten']
newList=[mydict[k] for k in mykeys if k in mydict]
print newList
[3, 1]답변
이 시도:
mydict = {'one': 1, 'two': 2, 'three': 3}
mykeys = ['three', 'one'] # if there are many keys, use a set
[mydict[k] for k in mykeys]
=> [3, 1]답변
new_dict = {x: v for x, v in mydict.items() if x in mykeys}