使用字典理解将np.nan插入字典元素

问题描述

我想知道如何替换字典中不符合条件的元素。例如,我要用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循环是如何填充字典的,而第二个是如何处理列表的内容的。