如果数组中的所有变量都匹配另一个数组中的变量，如何退出 While 循环 (python)

Question

如果数组的列中的所有值都与另一个数组的列匹配，我尝试进入/退出 while 循环。

例如，我有；

import numpy as np

ColumnA = np.array([[1], [1], [1], [-1], [-1], [1]])
ColumnB = np.array([[1], [1], [1], [-1], [-1], [-5]])

iterations = 10
count = 0

while iterations != count:
    count = count + 1
    print (count)
    while ColumnA.all() != ColumnB.all():
        ColumnB = ColumnB[0, 5] + 1
        print(ColumnB)

print("Exit")
print(count)

我期望这段代码进入嵌套的 while 循环，因为 ColumnA[0, 5] 不等于 ColumnB[0, 5]，然后 ColumnB[0, 5] 加 1 直到匹配 ColumnA[ 0, 5] 此时退出 while 循环。

此代码不会进入 while 循环，因此我假设它认为 ColumnA.all() 与 ColumnB.all() 相同。我的猜测是它只是查看数组的第一个值，但我希望它匹配数组的所有部分以退出循环。

感谢您的帮助。

Answer 1

使用 ColumnA.all() 您只需检查 ColumnA 中的所有元素是否都计算为 true。 ColumnA 和 ColumnB 就是这种情况。因此，ColumnA.all() == ColumnB.all() 的计算结果为 True == True，这当然是正确的。您需要按元素进行比较，然后检查获得的布尔向量的所有元素是否为 True：

while not all(ColumnA == ColumnB):
    ...

此外您的索引也是错误的。您的数组是 6x1，因此如果您想要最后一个元素，则需要 [5, 0] 而不是 [0, 5]。