我有一个字典列表。对于每个字典,我需要将其属性之一传递给函数,然后根据该函数的结果分配一个新属性。
例如:
我有一个网站的页面列表。我需要遍历它们,并根据URL将作者名称分配给dict中的属性。
for index, page in enumerate(pages):
pages[index]['author'] = get_author(page['url'])
这可行。但是很混乱,并且感觉不到pythonic。
pages[index]
感觉就像我在Python中不需要做的事情。有没有办法通过列表理解做到这一点?还是其他一些pythonic方式?
pages = [??? for page in pages]
最佳答案
您可以使用这样的list
理解:
result = [{**page, 'author': get_author(page['url'])}
for page in pages]
# This works too:
result = [dict(**page, author=get_author(page['url']))
for page in pages]
# but is less preferred because it will fail for input containing non-string keys
这将基于应用于与
dict
密钥对应的值的dict
的值,为每个带有附加键author
的原始get_author
创建一个新的url
。请注意,它不会修改原始的
list
。例:
def get_author(i):
if i == 1:
return 'hello'
else:
return 'bye'
pages = [{'url': 1},
{'url': 2}]
result = [{**page, **{'author': get_author(page['url'])}} for page in pages]
print(result)
输出:
[{'url': 1, 'author': 'hello'}, {'url': 2, 'author': 'bye'}]