python - 根据键值过滤Python字典

Question

我有一个字典dictM，其形式为

dictM={movieID:[rating1,rating2,rating3,rating4]}

Key 是 movieID， rating1、 rating2、 rating3、 rating4 是其值。有几个带有评级的 movieID。如果某个 movieID 具有一定数量的评分，我想将某些 movieID 连同评分一起移动到新字典中。

我正在做的是:

for movie in dictM.keys():
    if len(dictM[movie])>=5:
        dF[movie]=d[movie]

但我没有得到想要的结果。有人知道这个问题的解决方案吗？

Answer 1

您可以使用dictionary comprehension ，如下:

>>> dictM = {1: [1, 2, 3, 4], 2: [1, 2, 3]}

>>> {k: v for (k, v) in dictM.items() if len(v) ==4}
{1: [1, 2, 3, 4]}