主页 / 问题 / 1551719
Accepted
denisnumb
Asked:2023-11-17 02:49:13 +0000 UTC2023-11-17 02:49:13 +0000 UTC

Python：检查元素是否在字典中并使用它的哪种选项在性能方面更好？

  • 772

有一个字典，其结构如下：

data = {
    'key': {
        'subkey': 'value'
    }
}

并且需要获取嵌套字典的值，前提是键的存在'key'未知data。

可以这样写：

value = None
if 'key' in data:
    value = data['key']['subkey']

或者排成一行。我经常使用2种方法。

第一的：

value = None if 'key' not in data else data['key']['subkey']

第二：

value = None if not (subdict := data.get('key')) else subdict['subkey']

问题是：在短时间内多次执行此类检查的情况下，使用哪个选项有区别吗？也许有一些更简单的方法？

python

2 个回答

  1. data = {
    'key': {
        'subkey': 'value'
    }
}
print(data.get('key') and data['key']['subkey'])

    修改选项 cin比 c 稍快get()，但如果失败，则返回 False 而不是 None：

    'key' in d and d['key']['subkey']
    • 2
  2. Best Answer

    页面上提供的选项比较：

    from time import time

data = [{
    'key': {
        'subkey': 'value'
    }
}, {
    'key1': {
        'subkey': 'value'
    }
}]


def f1(d):
    value = None
    if 'key' in d:
        return d['key']['subkey']


def f2(d):
    return None if 'key' not in d else d['key']['subkey']


def f3(d):
    return None if not (subdict := d.get('key')) else subdict['subkey']


def f4(d):
    return d.get('key', {'subkey': None})['subkey']


def f5(d):
    return d.get('key') and d['key']['subkey']


n = 10**7

for i in f1, f2, f3, f4, f5:
    t = time()
    for j in range(n):
        i(data[0])
        i(data[1])
    print(f'Время варианта {i.__name__} = {time() - t}')

    Время варианта f1 = 2.34299898147583
Время варианта f2 = 2.1999993324279785
Время варианта f3 = 2.9900007247924805
Время варианта f4 = 3.6739983558654785
Время варианта f5 = 2.582000494003296
    • 2

相关问题