问题优化是否在循环中

Question

我有一个小问题，如果循环中的条件有所不同。我确定有一种优化此代码的方法，但我不知道如何...非常感谢您的帮助！

list1 = list()
list2 = list()
for item1, item2 in zip(data1, data2):
    if 'link' in item1 and 'link' in item2:
        list1.append(item1['link'])
        list2.append(item2['link'])
    elif 'link' in item1['details'] and 'link' in item2['details']:
        list1.append(item1['details']['link'])
        list2.append(item2['details']['link'])
    elif 'title' in item1 and 'title' in item2:
        list1.append(item1['title'])
        list2.append(item2['title'])
    elif 'description' in item1 and 'description' in item2:
        list1.append(item1['description'])
        list2.append(item2['description'])
    elif 'title' in item1['nav']['side'] and 'title' in item2['nav']['side']:
        list1.append(item1['nav']['side']['title'])
        list2.append(item2['nav']['side']['title'])
    elif 'title' in item1['nav']['top'] and 'title' in item2['nav']['top']:
        list1.append(item1['nav']['top']['title'])
        list2.append(item2['nav']['top']['title'])

Answer 1

您的问题不是很清楚，因为您不给我们输入数据。您可以使用Python Get做这样的事情。如果不在dict中，则python返回无。Python或短路，因此第一个不是没有返回的项目。这将解决您的问题。

list1 = list()
list2 = list()
for item1, item2 in zip(data1, data2):
    if 'link' in item1 and 'link' in item2:
        item1_to_append = item1.get('link') 
                        or item1.get('details', {}).get('link') 
                        or item1.get('title', {}) 
                        or item1.get('description', {}) 
                        or item1.get('nav', {}).get('side', {}).get('title') 
                        or item1.get('nav', {}).get('top', {}).get('title')
        item2_to_append = item2.get('link') 
                        or item2.get('details', {}).get('link') 
                        or item2.get('title', {}) 
                        or item2.get('description', {}) 
                        or item2.get('nav', {}).get('side', {}).get('title') 
                        or item2.get('nav', {}).get('top', {}).get('title')
        if item1_to_append:
            list1.append(item_to_append)
        if item2_to_append: 
            list2.append(item2_to_append)