问题描述
我想知道如何替换字典中不符合条件的元素。例如,我要用nan替换字典中所有不等于2的元素:
test = {'a': [1,2,3,4],'b': [2,4,5]}
我正在使用以下代码:
test2 = {name: item if item == 2 else np.nan for name in test.keys() for item in test[name]}
我想要以下答案:
test2 = {'a': [nan,nan,nan],nan]}
相反,我得到:
test2 = {'a': nan,'b': nan}
例如,如果我通过列表理解来做到这一点,例如
a = [1,4]
a2 = [item if item == 2 else np.nan for item in a]
a2 = [nan,nan]
我不确定我要去哪里哪里?
谢谢!
解决方法
您要嵌套循环,但未指定要以列表为目标。他们都将相同的键添加到同一词典中,从而不断覆盖自己。要解决此问题,请执行以下操作:
["bridge:length"=100]
请注意第一个for循环是如何填充字典的,而第二个是如何处理列表的内容的。