python - numpy.linspace 浮点数不准确

Question

有人可以解释一下，这是怎么回事？如何正确/稳健地检查数组中是否包含某些内容的最佳实践是什么？

Python 3.9.6 (default, Jun 30 2021, 10:22:16) 
[GCC 11.1.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy as np
>>> np.__version__
'1.20.3'

>>> np.linspace(0,1,11)
array([0. , 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1. ])

>>> 0.5 in np.linspace(0,1,11)
True
>>> 0.6 in np.linspace(0,1,11)
False

>>> np.isin(0.6, np.linspace(0,1,11))
array(False)

>>> 0.6 in np.round(np.linspace(0,1,11), 1)
True

标签： pythonnumpyroundingprecision

Answer 1

如果您不必那么精确，您可以使用类似np.isclose. 如果需要，您可以提供公差值。
例如：

>>> a = np.linspace(0,1,11)
>>> any(np.isclose(a,.6))
True