我有一个字典列表。对于每个字典,我需要将其属性之一传递给函数,然后根据该函数的结果分配一个新属性。

例如:
我有一个网站的页面列表。我需要遍历它们,并根据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'}]

09-30 14:06
查看更多