大整数ID匹配

ID_potential 到 numpy 阵列:

>>> np.flatnonzero(np.asarray(ID_potential) == ID_target).tolist()
[2, 5, 7]

另一个版本:

>>> np.where(np.equal(ID_potential, ID_target))[0].tolist()

如果这在您的实际代码中不起作用,那么很可能存在类型不匹配 ID\u电位 ID_target 在实际代码/数据中,或某些数据是浮点数据,因此严格的等式可能不成立。在这种情况下,使用 numpy.isclose() .

为了使代码更容易出错,请尝试以下版本:

np.where(np.equal(np.asarray(ID_potential, dtype=np.int), np.int(ID_target)))[0].tolist()

或者,如果值实际上是 float -更换 np.int 上面带有 np.float 和 np.equal 具有 np.isclose() :

np.where(np.isclose(np.asarray(ID_potential, dtype=np.float), np.float(ID_target)))[0].tolist()