python - 如果 X 和 Y 匹配 NumPy 数组中的元素

Question

如何检查数组中是否有一个元素的两个轴 [X, Y] 匹配 [drawx, drawy]？

我有一个 NumPy 数组:

#format: [X, Y]
wallxy = numpy.array([[0,1],[3,2],[4,6]])

和另外两个变量:

#attached to a counter that increases value each loop
drawx = 3
drawy = 2

我将数组用作一组位置 [[0,1],[3,2],[4,6]] 我需要测试 [ drawx, drawy] (也代表一个位置)在X轴和Y轴等的位置之一。 drawx = 4 drawy = 6 返回true drawx = 3 drawy = 2 返回 true drawx = 4 drawy = 2 返回 false drawx = 2 drawy = 1 返回 false

Answer 1

== 将广播比较结果，所以

wallxy = numpy.array([[0, 1],[3, 2][4, 6]])
z0 = numpy.array([3,2])
z1 = numpy.array([2,3])

(z0==wallxy).all(1).any()  # True
(z1==wallxy).all(1).any()  # False

我认为这就是您要找的东西。

打印出中间步骤将有助于理解和完成类似的任务:

z0 == wallxy     # checks which elements match
#  array([[False, False],
#         [ True,  True],
#         [False, False]], dtype=bool)

(z0==wallxy).all(1)   # checks whether all elements of axis 1 match
#  array([False,  True, False], dtype=bool)

(z0==wallxy).all(1).any()   # checks whether any axis 1 matches z0
#  True

如果您改为使用 z0 = numpy.array([2,3])，那么一切都会是 False。