python dict[str, list[str]] 问题 图片已经给出

就是这样一个dict[str, list[str]]
怎么写出一个function 让我可以select出数值小于一定数值的 并且保留两个column
就比如说我只要小于四的结果
就最后出来的结果得是
{‘comp_major’: ['No', 'No', 'No', 'No'], difficulty": ['3', '2', '3', '2']

你题目的解答代码如下：

def fun(d,n):
    li = zip(*d.values())
    li = [x for x in li if x[1]<n]
    for i,k in enumerate(d.keys()):
        d[k] = [x[i] for x in li]

dic = {
    'comp_major': ['a','b','c','d','e','f','g'],
    'difficulty': [2,7,5,3,9,4,1]
}

fun(dic,4)
print(dic)

如有帮助，望采纳！谢谢!