【发布时间】:2020-05-07 00:54:37
【问题描述】:
我在使用 pytorch 从自定义数据集创建的列表中调用了列表。
In [1]: sample[1][1]
Out[1]: [0.7, nan, nan, nan, nan]
我想检查列表中是否有 nan 值。我试过了
In [2]: np.nan in sample[1][1]
Out[2]: False
但它返回的是 False 输出。检查其中一个元素,我发现:
In [3]: sample[1][1][4] == np.nan
Out[3]: False
据推测,它一定是真的,因为检查它的类型会提示我:
In [4]: type(sample[1][1][4])
Out[4]: numpy.float64
需要帮助看看我哪里出错了。我希望该列表返回 True,因为它包含一个 nan。任何线索都会有所帮助
【问题讨论】:
-
你应该使用
np.isnan(sample) -
sample[1][1]的类型是什么?列表不会发生这种情况。