对列表中的列表进行列表理解

时间:2013-10-25 16:17:36

标签: python

如果我创建两个包含这样的列表的列表:

bad_list.append(['blue_widget', 'cracked', '776'])
bad_list.append(['red_widget', 'not_smooth', '545']) 
bad_list.append(['yellow_widget', 'spots', '35']) 
bad_list.append(['green_widget', 'smells_bad', '10'])
bad_list.append(['purple_widget', 'not_really_purple', '10'])


good_list.append(['blue_widget', 'ok', '776'])
good_list.append(['red_widget', 'ok', '545']) 
good_list.append(['green_widget', 'ok', '10'])

我希望能够使用列表理解来比较两个列表并删除 坏列表中使用第一个元素出现在好列表中的所有项目 (x_widget)作为要比较的项目。使用上面的例子,我应该留下:

['yellow_widget', 'spots', '35']
['purple_widget', 'not_really_purple', '10']

我尝试使用列表理解并且它可以工作,但新列表不会保留每一行:

final_list = [x for x in bad_list[0] if x not in good_list[0]]

当我使用final_list中的项目打印出内容时,我会得到类似的内容:

yellow_widget
smells_bad
10

任何线索都会非常感激。

5 个答案:

答案 0 :(得分:1)

一个班轮

[x for x in bad_list if any(x[0] == y[0] for y in good_list)]

*感谢@Bakuriu

答案 1 :(得分:0)

没有真正优化,但这应该有效:http://codecube.io/AD7RHA

bad_list=[]
good_list=[]

bad_list.append(['blue_widget', 'cracked', '776'])
bad_list.append(['red_widget', 'not_smooth', '545']) 
bad_list.append(['yellow_widget', 'spots', '35']) 
bad_list.append(['green_widget', 'smells_bad', '10'])
bad_list.append(['purple_widget', 'not_really_purple', '10'])


good_list.append(['blue_widget', 'ok', '776'])
good_list.append(['red_widget', 'ok', '545']) 
good_list.append(['green_widget', 'ok', '10'])

# ['yellow_widget', 'spots', '35']
# ['purple_widget', 'not_really_purple', '10']

labels = zip(*good_list)[0]

new_bad_list=[]

for item in bad_list:
    if item[0] not in labels:
        new_bad_list.append(item)

print new_bad_list

或这个单行:

new_bad_list=[item for item in bad_list if item[0] not in zip(*good_list)[0]]

答案 2 :(得分:0)

试试这个:

print [ele for ele in bad_list if ele[0] not in [i[0] for i in good_list]]

输出:

[['yellow_widget', 'spots', '35'], ['purple_widget', 'not_really_purple', '10']]

答案 3 :(得分:0)

有一种更有效的解决方案。从列表中创建一个集合

bad_set = set(bad_list)
good_set = set(good_list)

现在要删除好列表中存在的错误列表中的所有项目,您可以简单地减去这些集合:

bad_set - good_set

如果您愿意,可以将设置转换回列表。

答案 4 :(得分:0)

最简单的方法是:

final_list = [x for x in bad_list if x[0] not in [x[0] for x in good_list]]

但请注意,测试列表中存在的元素效率不高。

所以,你可以先建立一个集合:

good_list_names = set([x[0] for x in good_list])

然后:

final_list = [x for x in bad_list if x[0] not in good_list_names]