Витягнути підмножину пар ключ-значення з об’єкта словника Python?


313

У мене великий об’єкт словника, який має кілька пар ключових значень (близько 16), але мене цікавлять лише 3 з них. Який найкращий спосіб (найкоротший / ефективніший / найелегантніший) досягти цього?

Найкраще, що я знаю:

bigdict = {'a':1,'b':2,....,'z':26} 
subdict = {'l':bigdict['l'], 'm':bigdict['m'], 'n':bigdict['n']}

Я впевнений, що є більш елегантний спосіб, ніж цей. Ідеї?

Відповіді:


430

Ви можете спробувати:

dict((k, bigdict[k]) for k in ('l', 'm', 'n'))

... або в Пітон 3Версії Python 2.7 або новішої версії (спасибі Фабіо Дініз, який вказав, що він працює і в 2.7) :

{k: bigdict[k] for k in ('l', 'm', 'n')}

Оновлення: Як зазначає Håvard S , я припускаю, що ви знаєте, що ключі будуть у словнику - дивіться його відповідь, якщо ви не можете зробити це припущення. Крім того, як в коментарях вказує timbo , якщо ви хочете, щоб ключ, який не вистачає в bigdictкарті None, можна зробити:

{k: bigdict.get(k, None) for k in ('l', 'm', 'n')}

Якщо ви використовуєте Python 3, і вам потрібні лише ключі в новому диктаті, які фактично існують у вихідному, ви можете використовувати факт для перегляду об’єктів, що реалізують деякі задані операції:

{k: bigdict[k] for k in bigdict.keys() & {'l', 'm', 'n'}}

5
Не вдасться, якщо bigdictне міститьk
Håvard S

7
Трохи суворо підкресливши це - мені здавалося досить зрозумілим з контексту, що відомо, що ці ключі є у словнику ...
Марк Лонгейр

9
{k: bigdict.get(k,None) for k in ('l', 'm', 'n')}розберемось із ситуацією, коли вказаний ключ відсутній у словнику джерела, встановивши ключ у новому
диктаті

9
@MarkLongair Залежно від випадку використання {k: bigdict [k] для k in ('l', 'm', 'n'), якщо k у bigdict} може бути краще, оскільки він зберігає лише ключі, які насправді мають значення.
Бріфорд Вілі

6
bigdict.keys() & {'l', 'm', 'n'} ==> bigdict.viewkeys() & {'l', 'm', 'n'} для Python2.7
kxr

119

Трохи коротше, принаймні:

wanted_keys = ['l', 'm', 'n'] # The keys you want
dict((k, bigdict[k]) for k in wanted_keys if k in bigdict)

8
+1 для альтернативної поведінки виключення ключа, якщо він не є виразним, а не встановити його на "None".
dhj

1
Як варіант: dict((k,bigdict.get(k,defaultVal) for k in wanted_keys)якщо у вас повинні бути всі ключі.
Томас Ендрюс

2
Ця відповідь зберігається за допомогою "t".
сакурашинкен


16

Трохи порівняння швидкості для всіх згаданих методів:

Python 2.7.11 |Anaconda 2.4.1 (64-bit)| (default, Jan 29 2016, 14:26:21) [MSC v.1500 64 bit (AMD64)] on win32
In[2]: import numpy.random as nprnd
keys = nprnd.randint(1000, size=10000)
bigdict = dict([(_, nprnd.rand()) for _ in range(1000)])

%timeit {key:bigdict[key] for key in keys}
%timeit dict((key, bigdict[key]) for key in keys)
%timeit dict(map(lambda k: (k, bigdict[k]), keys))
%timeit dict(filter(lambda i:i[0] in keys, bigdict.items()))
%timeit {key:value for key, value in bigdict.items() if key in keys}
100 loops, best of 3: 3.09 ms per loop
100 loops, best of 3: 3.72 ms per loop
100 loops, best of 3: 6.63 ms per loop
10 loops, best of 3: 20.3 ms per loop
100 loops, best of 3: 20.6 ms per loop

Як і очікувалося: розуміння словника - найкращий варіант.


Перші 3 операції роблять іншу дію, ніж останні дві, і приведе до помилки, якщо її keyнемає в bigdict.
naught101

12

Ця відповідь використовує розуміння словника, подібне до вибраної відповіді, але не буде винятком випадків, коли вони відсутні.

версія python 2:

{k:v for k, v in bigDict.iteritems() if k in ('l', 'm', 'n')}

версія python 3:

{k:v for k, v in bigDict.items() if k in ('l', 'm', 'n')}

2
... але якщо великий диктат є ВЕЛИЧЕЗНИМ, він все одно буде повторений повністю (це операція O (n)), тоді як обернена просто захопить 3 елементи (кожна операція O (1)).
wouter bolsterlee

1
Питання стосується словника всього 16 клавіш
Meow

6

Можливо:

subdict=dict([(x,bigdict[x]) for x in ['l', 'm', 'n']])

Python 3 навіть підтримує наступне:

subdict={a:bigdict[a] for a in ['l','m','n']}

Зауважте, що ви можете перевірити наявність у словнику наступним чином:

subdict=dict([(x,bigdict[x]) for x in ['l', 'm', 'n'] if x in bigdict])

респ. для пітона 3

subdict={a:bigdict[a] for a in ['l','m','n'] if a in bigdict}

Якщо aне,bigdict
Håvard S

3

Гаразд, це щось, що мене турбувало кілька разів, тож дякую Джаєш за те, що просив.

Наведені вище відповіді здаються настільки ж хорошим рішенням, як і будь-яке, але якщо ви використовуєте це у всьому коді, має сенс перетворити функціональність IMHO. Також тут є два можливі випадки використання: той, де ви дбаєте про те, чи всі ключові слова є в оригінальному словнику. і туди, де тебе немає. Було б непогано ставитися до обох однаково.

Отже, для моєї вартості на дві копійки пропоную написати підклас словника, наприклад

class my_dict(dict):
    def subdict(self, keywords, fragile=False):
        d = {}
        for k in keywords:
            try:
                d[k] = self[k]
            except KeyError:
                if fragile:
                    raise
        return d

Тепер ви можете витягнути під-словник із

orig_dict.subdict(keywords)

Приклади використання:

#
## our keywords are letters of the alphabet
keywords = 'abcdefghijklmnopqrstuvwxyz'
#
## our dictionary maps letters to their index
d = my_dict([(k,i) for i,k in enumerate(keywords)])
print('Original dictionary:\n%r\n\n' % (d,))
#
## constructing a sub-dictionary with good keywords
oddkeywords = keywords[::2]
subd = d.subdict(oddkeywords)
print('Dictionary from odd numbered keys:\n%r\n\n' % (subd,))
#
## constructing a sub-dictionary with mixture of good and bad keywords
somebadkeywords = keywords[1::2] + 'A'
try:
    subd2 = d.subdict(somebadkeywords)
    print("We shouldn't see this message")
except KeyError:
    print("subd2 construction fails:")
    print("\toriginal dictionary doesn't contain some keys\n\n")
#
## Trying again with fragile set to false
try:
    subd3 = d.subdict(somebadkeywords, fragile=False)
    print('Dictionary constructed using some bad keys:\n%r\n\n' % (subd3,))
except KeyError:
    print("We shouldn't see this message")

Якщо ви запускаєте всі наведені вище коди, ви повинні побачити (щось на кшталт) наступного виводу (вибачте за форматування):

Оригінальний словник:
{'a': 0, 'c': 2, 'b': 1, 'e': 4, 'd': 3, 'g': 6, 'f': 5, 'i': 8, 'h': 7, 'k': 10, 'j': 9, 'm': 12, 'l': 11, 'o': 14, 'n': 13, 'q': 16, 'p': 15, 's': 18, 'r': 17, 'u': 20, 't': 19, 'w': 22, 'v': 21, 'y': 24, 'x ': 23,' z ': 25}

Словник із непарних нумерованих клавіш:
{'a': 0, 'c': 2, 'e': 4, 'g': 6, 'i': 8, 'k': 10, 'm': 12, ' o ': 14,' q ': 16,' s ': 18,' u ': 20,' w ': 22,' y ': 24}

не вдалося побудувати subd2:
оригінальний словник не містить деяких клавіш

Словник, побудований за допомогою поганих клавіш:
{'b': 1, 'd': 3, 'f': 5, 'h': 7, 'j': 9, 'l': 11, 'n': 13, 'p': 15, 'r': 17, 't': 19, 'v': 21, 'x': 23, 'z': 25}


1
Підкласифікація вимагає перетворення існуючого об'єкта dict у тип підкласу, що може бути дорогим. Чому б просто не написати просту функцію subdict(orig_dict, keys, …)?
musiphil

3

Ви також можете скористатися map(що дуже корисна функція для ознайомлення в будь-якому випадку):

sd = dict(map(lambda k: (k, l.get(k, None)), l))

Приклад:

large_dictionary = {'a1':123, 'a2':45, 'a3':344}
list_of_keys = ['a1', 'a3']
small_dictionary = dict(map(lambda key: (key, large_dictionary.get(key, None)), list_of_keys))

PS: Я запозичив .get(key, None)попередню відповідь :)


1

Ще одна (я вважаю за краще відповідь Марка Лонгайра)

di = {'a':1,'b':2,'c':3}
req = ['a','c','w']
dict([i for i in di.iteritems() if i[0] in di and i[0] in req])

його повільний для великого диктує - х
KXR

0

рішення

from operator import itemgetter
from typing import List, Dict, Union


def subdict(d: Union[Dict, List], columns: List[str]) -> Union[Dict, List[Dict]]:
    """Return a dict or list of dicts with subset of 
    columns from the d argument.
    """
    getter = itemgetter(*columns)

    if isinstance(d, list):
        result = []
        for subset in map(getter, d):
            record = dict(zip(columns, subset))
            result.append(record)
        return result
    elif isinstance(d, dict):
        return dict(zip(columns, getter(d)))

    raise ValueError('Unsupported type for `d`')

приклади використання

# pure dict

d = dict(a=1, b=2, c=3)
print(subdict(d, ['a', 'c']))

>>> In [5]: {'a': 1, 'c': 3}
# list of dicts

d = [
    dict(a=1, b=2, c=3),
    dict(a=2, b=4, c=6),
    dict(a=4, b=8, c=12),
]

print(subdict(d, ['a', 'c']))

>>> In [5]: [{'a': 1, 'c': 3}, {'a': 2, 'c': 6}, {'a': 4, 'c': 12}]
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.