使用字典理解将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循环是如何填充字典的，而第二个是如何处理列表的内容的。