numpy-ndarray - np.where() 与 np.linspace()

Question

我正在尝试使用 np.where() 在 x_norm 数组中查找元素的索引，但效果不佳。有没有办法找到元素的索引？

x_norm  = np.linspace(-10,10,1000)
np.where(x_norm == -0.19019019)

Np.where 与 np.arange() 一起使用，可以找到由 linspace 创建的数组的第一个或最后一个元素的索引。

Answer 1

生成的数字np.linspace包含的小数位数比您粘贴到 np.where (-0.19019019019019012) 的数字多。

因此，最好使用np.argmin找到最接近的值并避免舍入错误：

x_norm  = np.linspace(-10,10,1000)
yournumber=-0.19019019
idx=np.argmin(np.abs(x_norm-yournumber))

然后，您可以进一步添加np.where(x_norm==x_norm[idx])到您的代码中，以防您有重复的数组。

Answer 2

使用 np.round 将精度级别设置为 8，然后使用 np.where 将数据过滤为掩码，然后将掩码应用于数组。

x_norm  = np.round(np.asarray(np.linspace(-10,10,1000)),8)
results=x_norm[np.where(x_norm==-9.91991992)]
print(results)