从列表列表中删除列表

时间:2015-01-04 20:13:35

标签: python list python-2.7

我想循环遍历字典中的列表列表,并根据列表中的值等于None删除某些列表。我的代码如下。它不是删除符合条件的列表,而是始终删除第一个列表。

我的代码:

My_Table = {
    'Table1' : {'Results': [["pineapple", 5], ["cherry", 7], ["kumquat", None], ["plum", None]]},
    'Table2' : {'Results': [["Orange", 5], ["Apple", 7], ["Pear", None], ["cantelope", None]]},
}

for tbl, values in My_Table.viewitems():
      for item_values in values['Results']:
        if item_values[1] is not None:
            values['Results'].remove(item_values)
      print values['Results']

输出:

[['Apple', 7], ['Pear', None], ['cantelope', None]]
[['cherry', 7], ['kumquat', None], ['plum', None]]

期望的输出:

[['Orange', 5], ['Apple', 7]]
[['pineapple', 5], ['cherry', 7]]

1 个答案:

答案 0 :(得分:3)

在迭代序列时,切勿从序列中删除项目。这样做会搞砸Python的迭代协议,并经常导致它意外跳过项目。

相反,您可以使用list comprehension和切片分配:

for tbl, values in My_Table.viewitems():
    values['Results'][:] = [x for x in values['Results'] if x[1] is not None]
    print values['Results']

演示:

>>> My_Table = {
...     'Table1' : {'Results': [["pineapple", 5], ["cherry", 7], ["kumquat", None], ["plum", None]]},
...     'Table2' : {'Results': [["Orange", 5], ["Apple", 7], ["Pear", None], ["cantelope", None]]},
... }
>>> for tbl, values in My_Table.viewitems():
...      values['Results'][:] = [x for x in values['Results'] if x[1] is not None]
...      print values['Results']
...
[['Orange', 5], ['Apple', 7]]
[['pineapple', 5], ['cherry', 7]]
>>>

与您当前的代码不同,这将首先构建一个新列表,其中包含我们要保留的所有项目。之后,它将使用切片分配将这些项目传输到旧列表中。

[:]使其成为就地操作。意思是,values['Results']对象仅被修改,而不是被替换。如果我们这样做了:

values['Results'] = [x for x in values['Results'] if x[1] is not None]

代码仍然有效,但列表将替换为理解中的新代码。