Python:同时循环遍历字典中的所有列表以过滤出元素

时间:2018-12-11 19:56:22

标签: python python-3.x list dictionary

我有一些字典

someDict = {
    'foo1': [1, 4, 7, 0, -2],
    'foo2': [0, 2, 5, 3, 6],
    'foo3': [1, 2, 3, 4, 5]
}

我想使用Python 3遍历每个列表中的所有元素,当给定索引处的元素等于零时,我想删除列表中 all 处该索引处的该元素/字典中的属性。这样字典以

结尾
someDict = {
    'foo1': [4, 7, -2],
    'foo2': [2, 5, 6],
    'foo3': [2, 3, 5]
}

请注意,我事先不知道字典将包含多少个键/列表,并且我也不知道列表将包含多少个元素。我想出了下面的代码,该代码似乎可行,但是想知道是否有更有效的方法?

keyPropList = someDict.items()
totalList = []

for tupleElement in keyPropList:
    totalList.append(tupleElement[1])

copyTotalList = totalList[:]
for outerRow in copyTotalList:
    for outerIndex, outerElement in enumerate(outerRow):
        if outerElement==0:
            for innerIndex, _ in enumerate(copyTotalList):
                del totalList[innerIndex][outerIndex]

print('someDict =', someDict)

3 个答案:

答案 0 :(得分:8)

您可以找到“被禁止”索引的列表,这些列表随后可用于过滤结构:

someDict = {
  'foo1': [1, 4, 7, 0, -2],
  'foo2': [0, 2, 5, 3, 6],
  'foo3': [1, 2, 3, 4, 5]
}
results = {i for c in someDict.values() for i, a in enumerate(c) if not a}
new_dict = {a:[c for i, c in enumerate(b) if i not in results] for a, b in someDict.items()}

输出:

{'foo1': [4, 7, -2], 'foo2': [2, 5, 6], 'foo3': [2, 3, 5]}

答案 1 :(得分:3)

这是一个厚脸皮的单线纸:

>>> dict(zip(someDict, map(list, zip(*(ts for ts in zip(*someDict.values()) if all(ts))))))
{'foo3': [2, 3, 5], 'foo1': [4, 7, -2], 'foo2': [2, 5, 6]}
>>>

答案 2 :(得分:0)

其他选项,pop的详细程度和破坏性(original_dict):

some_dict = {
    'foo1': [1, 0, 2, 3, -2],
    'foo2': [0, 2, 5, 3, 6],
    'foo3': [1, 2, 3, 0, 5]
}

for k, v in some_dict.items():
  id = []
  for i, e in enumerate(v):
    if e == 0: id.append(i-len(id))
  [ [ some_dict[k].pop(x) for x in id ] for k,v in some_dict.items() ]

print(some_dict)
#=> {'foo1': [2, -2], 'foo2': [5, 6], 'foo3': [3, 5]}