python - 使用多个键之一过滤字典

Question

我是 Pyhton 的新手，想用由两个值组成的键来过滤字典。这是我的字典:

{(0, 'DRYER'): [103.0, 131.0, 9.0, 1.24],
 (2, 'DRYER'): [106.0, 120.0, 5.0, 1.24],
 (2, 'WASHING'): [70.0, 90.0, 11.0, 0.19]}

键由两个值 (n,a) 组成我想创建一个新字典，其中 n=2 ，结果是:

{(2, 'DRYER'): [106.0, 120.0, 5.0, 1.24],
 (2, 'WASHING'): [70.0, 90.0, 11.0, 0.19]}

如果有人能帮忙，我将不胜感激!

Answer 1

使用字典理解检查元组键的第一个值是否为 2:

{k: v for k, v in dct.items() if k[0]==2}

所以:

In [11]: dct = {(0, 'DRYER'): [103.0, 131.0, 9.0, 1.24],
    ...:  (2, 'DRYER'): [106.0, 120.0, 5.0, 1.24],
    ...:  (2, 'WASHING'): [70.0, 90.0, 11.0, 0.19]}
    ...:  

In [12]: {k: v for k, v in dct.items() if k[0]==2}
Out[12]: 
{(2, 'DRYER'): [106.0, 120.0, 5.0, 1.24],
 (2, 'WASHING'): [70.0, 90.0, 11.0, 0.19]}