在具有浮点值的二维数组中查找数组

Question

我有一个 numpy 数组，我想搜索它以找到一个数组（不是一个值）。

values = np.array([[0.73123909, 0.73298429, 0.73472949, 0.73647469, 1.        ],
                   [0.72949389, 0.46596859, 0.39441536, 0.87260035, 1.        ],
                   [0.2600349 , 0.05235602, 0.73298429, 0.96684119, 1.        ],
                   [0.83071553, 0.37172775, 0.7452007 , 0.08202443, 1.        ],
                   [0.27923211, 0.28097731, 0.28272251, 0.28446771, 1.        ]])

item = np.array([0.73123909, 0.73298429, 0.73472949, 0.73647469, 1.        ])
index = np.where(values == item)

我希望得到类似于index = 0 或(array([0]),) 的结果 但我明白了

(array([  0,   1,   2,   3,   4,]),
array([4, 4, 4, 4, 4 ]))

类似的答案是目标整数，不适用于浮点数。

Answer 1

试试这个：

>>> np.where(np.isclose(values, item).all(1))
(array([0], dtype=int64),)

顺便说一句，如果您正在处理浮点数，请不要使用 == 检查是否相等，而是使用 isclose。