Словари и значения по умолчанию - PullRequest
185 голосов
/ 20 февраля 2012

Предполагая, что connectionDetails - это словарь Python, какой самый лучший, самый элегантный, самый "питонический" способ рефакторинга кода подобен этому?

if "host" in connectionDetails:
    host = connectionDetails["host"]
else:
    host = someDefaultValue

Ответы [ 8 ]

273 голосов
/ 20 февраля 2012

Как это:

host = connectionDetails.get('host', someDefaultValue)
87 голосов
/ 29 мая 2014

Вы также можете использовать defaultdict примерно так:

from collections import defaultdict
a = defaultdict(lambda: "default", key="some_value")
a["blabla"] => "default"
a["key"] => "some_value"

Вы можете передать любую обычную функцию вместо лямбды:

from collections import defaultdict
def a():
  return 4

b = defaultdict(a, key="some_value")
b['absent'] => 4
b['key'] => "some_value"
24 голосов
/ 06 июля 2013

Хотя .get() идиома, она медленнее, чем if/else (и медленнее, чем try/except, если большую часть времени можно ожидать наличия ключа в словаре):

>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="try:\n a=d[1]\nexcept KeyError:\n a=10")
0.07691968797894333
>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="try:\n a=d[2]\nexcept KeyError:\n a=10")
0.4583777282275605
>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="a=d.get(1, 10)")
0.17784020746671558
>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="a=d.get(2, 10)")
0.17952161730158878
>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="if 1 in d:\n a=d[1]\nelse:\n a=10")
0.10071221458065338
>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}", 
... stmt="if 2 in d:\n a=d[2]\nelse:\n a=10")
0.06966537335119938
18 голосов
/ 06 июля 2013

Для нескольких различных значений по умолчанию попробуйте это:

connectionDetails = { "host": "www.example.com" }
defaults = { "host": "127.0.0.1", "port": 8080 }

completeDetails = {}
completeDetails.update(defaults)
completeDetails.update(connectionDetails)
completeDetails["host"]  # ==> "www.example.com"
completeDetails["port"]  # ==> 8080
7 голосов
/ 29 февраля 2016

В словарях python есть способ сделать это: dict.setdefault

connectionDetails.setdefault('host',someDefaultValue)
host = connectionDetails['host']

Однако этот метод устанавливает значение connectionDetails['host'] в someDefaultValue, если ключ host еще не определен, в отличие от того, что задан вопрос.

6 голосов
/ 27 апреля 2017

(это поздний ответ)

Альтернативой является создание подкласса класса dict и реализация метода __missing__(), например:

class ConnectionDetails(dict):
    def __missing__(self, key):
        if key == 'host':
            return "localhost"
        raise KeyError(key)

Примеры:

>>> connection_details = ConnectionDetails(port=80)

>>> connection_details['host']
'localhost'

>>> connection_details['port']
80

>>> connection_details['password']
Traceback (most recent call last):
  File "python", line 1, in <module>
  File "python", line 6, in __missing__
KeyError: 'password'
2 голосов
/ 18 июля 2017

Проверяя подозрение @Tim Pietzcker о ситуации в PyPy (5.2.0-alpha0) для Python 3.3.5, я обнаружил, что действительно и способ .get(), и способ if / else работают аналогично.На самом деле кажется, что в случае if / else есть даже только один поиск, если условие и присвоение включают в себя один и тот же ключ (сравните с последним случаем, когда есть два поиска).

>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="try:\n a=d[1]\nexcept KeyError:\n a=10")
0.011889292989508249
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="try:\n a=d[2]\nexcept KeyError:\n a=10")
0.07310474599944428
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="a=d.get(1, 10)")
0.010391917996457778
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="a=d.get(2, 10)")
0.009348208011942916
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="if 1 in d:\n a=d[1]\nelse:\n a=10")
0.011475925013655797
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="if 2 in d:\n a=d[2]\nelse:\n a=10")
0.009605801998986863
>>>> timeit.timeit(setup="d={1:2, 3:4, 5:6, 7:8, 9:0}",
.... stmt="if 2 in d:\n a=d[2]\nelse:\n a=d[1]")
0.017342638995614834
1 голос
/ 16 сентября 2017

Вы можете использовать функцию lamba для этого в качестве однострочника.Создайте новый объект connectionDetails2, доступ к которому осуществляется как функция ...

connectionDetails2 = lambda k: connectionDetails[k] if k in connectionDetails.keys() else "DEFAULT"

Теперь используйте

connectionDetails2(k)

вместо

connectionDetails[k]

, который возвращаетзначение словаря, если в ключах указано k, в противном случае возвращается "DEFAULT"

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...