NumPy数组真值歧义ValueError排查：int数组排序行时出错

问题场景 #

运行代码时触发错误：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

需求是：将numpy矩阵A中，每行-1（代表未知值）数量≤5的行，按-1数量从小到大排序后存入list_1，报错发生在调用list_1.index(item)获取数组索引时。

报错原因 #

list_1中存储的是numpy数组对象，当调用list_1.index(item)时，Python会尝试比较数组是否相等。numpy数组直接用==比较会返回一个布尔数组，Python无法直接判断这个布尔数组的整体真值（不知道是要所有元素都相等还是至少一个相等），因此抛出歧义错误。

修复方案 #

1. 避免使用list.index()方法：遍历list_1时用enumerate同时获取索引和元素，直接使用索引进行插入操作，跳过数组相等比较的步骤。
2. 简化-1数量计算：用numpy的向量运算替代嵌套循环，(行数组 == -1).sum()可以直接统计该行-1的数量，代码更简洁高效。

修复后的完整代码 #

import numpy as np
from colorama import Fore

list_2 = [[-1,9,4,2,-1,6,-1,-1,-1],
        [-1,-1,-1,-1,-1,4,-1,1,-1],
        [7,8,3,-1,-1,-1,2,-1,-1],
        [-1,4,3,-1,7,5,1,8,2],
        [8,5,6,-1,3,2,9,-1,4],
        [-1,-1,-1,-1,9,8,-1,6,-1],
        [-1,-1,7,8,-1,-1,-1,-1,-1],
        [-1,-1,-1,-1,-1,3,4,-1,7],
        [4,6,-1,-1,-1,-1,-1,3,1]]
A = np.array(list_2)

list_1 = []

print(Fore.RED, "Rows with less than 5 unknowns numbers:")
for i in range(9):
    # 用numpy向量运算统计当前行-1的数量
    S = (A[i] == -1).sum()
    if S <= 5:
        if len(list_1) > 0:
            append_last = True
            # 用enumerate同时获取索引和元素
            for idx, item in enumerate(list_1):
                S2 = (item == -1).sum()
                if S < S2:
                    list_1.insert(idx, A[i,:])
                    append_last = False
                    break
            if append_last:
                list_1.append(A[i,:])
        else:
            list_1.append(A[i,:])

print(Fore.WHITE, list_1)

额外优化建议 #

如果不需要手动实现排序逻辑，可以直接用numpy的排序功能，代码会更简洁：

# 计算每行-1的数量
counts = (A == -1).sum(axis=1)
# 筛选出-1数量≤5的行的索引
valid_indices = counts <= 5
# 按-1数量从小到大排序这些行
sorted_rows = A[valid_indices][np.argsort(counts[valid_indices])]
# 转换为list格式（如果需要）
list_1 = sorted_rows.tolist()

print(Fore.RED, "Rows with less than 5 unknowns numbers:")
print(Fore.WHITE, list_1)

内容的提问来源于stack exchange，提问作者Ernest Vandenbulcke