从一个字典中提取多个键：值对到一个新的字典

Question

我有一个包含一些数据的 dict 列表，我想将某些键：值对提取到一个新的 dict 列表中。 我知道我可以这样做的一种方法是使用del i['unwantedKey'] ，但是，我宁愿不删除任何数据，而是创建一个包含所需数据的新字典。

列顺序可能会改变，所以我需要一些东西从较大的字典中提取两个键：值对到一个新的字典中。

当前数据格式

[{'Speciality': 'Math', 'Name': 'Matt', 'Location': 'Miami'},
 {'Speciality': 'Science', 'Name': 'Ben', 'Location': 'Las Vegas'}, 
 {'Speciality': 'Language Arts', 'Name': 'Sarah', 'Location': 'Washington DC'},
 {'Speciality': 'Spanish', 'Name': 'Tom', 'Location': 'Denver'},
 {'Speciality': 'Chemistry', 'Name': 'Jim', 'Location': 'Dallas'}]

从字典中删除键：值的代码

import csv

data= []
for line in csv.DictReader(open('data.csv')):
    data.append(line)

for i in data:
    del i['Speciality']

print data

不使用del i['Speciality']所需的数据格式

[{'Name': 'Matt', 'Location': 'Miami'}, 
 {'Name': 'Ben', 'Location': 'Las Vegas'}, 
 {'Name': 'Sarah', 'Location': 'Washington DC'}, 
 {'Name': 'Tom', 'Location': 'Denver'}, 
 {'Name': 'Jim', 'Location': 'Dallas'}]

Answer 1

如果要提供肯定的键列表以复制到新词典中：

import csv

with open('data.csv', 'rb') as csv_file:
    data = list(csv.DictReader(csv_file))

keys = ['Name', 'Location']
new_data = [dict((k, d[k]) for k in keys) for d in data]

print new_data

Answer 2

假设我们有，

l1 = [{'Location': 'Miami', 'Name': 'Matt', 'Speciality': 'Math'},
      {'Location': 'Las Vegas', 'Name': 'Ben', 'Speciality': 'Science'},
      {'Location': 'Washington DC', 'Name': 'Sarah', 'Speciality': 'Language Arts'},
      {'Location': 'Denver', 'Name': 'Tom', 'Speciality': 'Spanish'},
      {'Location': 'Dallas', 'Name': 'Jim', 'Speciality': 'Chemistry'}]

创建不包含键“ Speciality”的新词典列表，

l2 = []
for oldd in l1:
    newd = {}
    for k,v in oldd.items():
        if k != 'Speciality':
            newd[k] = v
    l2.append(newd)

现在l2将是您想要的输出。 通常，您可以像这样排除任意键列表

exclude_keys = ['Speciality', 'Name']
l2 = []
    for oldd in l1:
        newd = {}
        for k,v in oldd.items():
            if k not in exclude_keys:
                newd[k] = v
        l2.append(newd)

使用include_keys变量也可以做到这一点

include_keys = ['Name', 'Location']
l2 = []
    for oldd in l1:
        newd = {}
        for k,v in oldd.items():
            if k in include_keys:
                newd[k] = v
        l2.append(newd)

Answer 3

您可以使用一行代码（Python 2.6+）创建新的字典列表，将其限制为所需的键：

NLoD=[{k:d[k] for k in ('Name', 'Location')} for d in LoD]

试试吧：

>>> LoD=[{'Speciality': 'Math', 'Name': 'Matt', 'Location': 'Miami'},
 {'Speciality': 'Science', 'Name': 'Ben', 'Location': 'Las Vegas'}, 
 {'Speciality': 'Language Arts', 'Name': 'Sarah', 'Location': 'Washington DC'},
 {'Speciality': 'Spanish', 'Name': 'Tom', 'Location': 'Denver'},
 {'Speciality': 'Chemistry', 'Name': 'Jim', 'Location': 'Dallas'}]
>>> [{k:d[k] for k in ('Name', 'Location')} for d in LoD]
[{'Name': 'Matt', 'Location': 'Miami'}, {'Name': 'Ben', 'Location': 'Las Vegas'}, {'Name': 'Sarah', 'Location': 'Washington DC'}, {'Name': 'Tom', 'Location': 'Denver'}, {'Name': 'Jim', 'Location': 'Dallas'}]

由于您使用的是csv，因此可以将首先读取的列限制为所需的列，这样就无需删除不需要的数据：

dc=('Name', 'Location')        
with open(fn) as f:
    reader=csv.DictReader(f)
    LoD=[{k:row[k] for k in dc} for row in reader]

Answer 4

keys_lst = [Name', 'Location]

new_data={key:val for key,val in event.items() if key in keys_lst}
print(new_data)