解析字典以在Python 3.6中检索密钥

Question

我有一个Python字典，我试图弄清楚如何获得一个特定的键和值。

这是示例Python字典，我需要检索category_id值。

lines = [ 
 {'id': 'sub_BUNbsaTbxzrZYW', 'category_id': 'prodcat_xMOTFxgQnA', 'object': 'line_item', 'amount': 9999, 'currency': 'usd', 'description': '1x Yearly (at $99.99)', 'discountable': True, 'livemode': True, 'metadata': {}, 'period': {'start': 1538833681, 'end': 1570369681}, 'plan': {'id': 'Nuts Yearly', 'object': 'plan', 'amount': 10000, 'created': 1498624603, 'currency': 'usd', 'interval': 'year', 'interval_count': 1, 'livemode': False, 'metadata': {}, 'name': 'Nuts Yearly', 'statement_descriptor': None, 'trial_period_days': None}, 'proration': False, 'quantity': 1, 'subscription': None, 'subscription_item': 'si_1B7OqTAQofPy1JZrjB5myHN5', 'type': 'subscription'}, 


 {'id': 'sub_BUNbsaTbxzrZYW', 'category_id': 'prodcat_jbWGPxLNHM', 'object': 'line_item', 'amount': 9999, 'currency': 'usd', 'description': '1x Yearly (at $99.99)', 'discountable': True, 'livemode': True, 'metadata': {}, 'period': {'start': 1538833681, 'end': 1570369681}, 'plan': {'id': 'Nuts Yearly', 'object': 'plan', 'amount': 10000, 'created': 1498624603, 'currency': 'usd', 'interval': 'year', 'interval_count': 1, 'livemode': False, 'metadata': {}, 'name': 'Nuts Yearly', 'statement_descriptor': None, 'trial_period_days': None}, 'proration': False, 'quantity': 1, 'subscription': None, 'subscription_item': 'si_1B7OqTAQofPy1JZrjB5myHN5', 'type': 'subscription'}], 'has_more': False, 'object': 'list', 'url': '/v1/invoices/in_1Bg1FZAQofPy1JZrLNlHERmz/lines'}] 

我可以使用以下方式获取数据：

cat_id = []
for i in lines:
    for k, v in i.items():
        if k == 'category_id':
            cat_id.append(v)

如何使我的代码在这种情况下更有效？

Answer 1

只需从字典中选择元素：

cat_id = []
for line in lines:
    cat_id.append(line['category_id'])

要么

cat_id = [line['category_id'] for line in lines]

Answer 2

如果您假设您的dicts的每个条目都包含该类别，则可以通过以下方式更快地执行此操作：

cat_id = []
for i in lines:
    cat_id.append(i.get("category_id"))

对于没有'category_id'的任何条目，'None'将保存到列表中