Python while循环处理numpy数组至空触发真值歧义报错的问题

• while eqvec的判空逻辑不适用numpy数组
  Python原生列表可以直接用while 列表名判断非空，但numpy数组做布尔值判断时，会因为数组包含多个元素产生歧义，没法直接得到“空数组为假、非空为真”的判断，这就是触发报错的直接原因。
• 对numpy数组误用列表方法
  你写的eqvec.remove(index_max)是Python原生列表的专属方法，numpy数组本身没有实现这个接口，就算解决了判空报错，执行到这行也会触发方法不存在的错误。而且remove方法的作用是删除列表中第一个和传入值相等的元素，不是按索引位置删除，你传入的是最大值对应的索引值，和数组存储的eqvec计算值根本不匹配，就算方法能用，删除逻辑也是错的。

方案1：匹配你原本的列表操作逻辑 #

把eqvec从numpy数组转成Python原生列表，适配你原本设计的“循环找最大值索引、删除对应元素直到清空”的流程：

import numpy as np

def ordering(n):
    # 把初始numpy零数组转成原生列表，后续操作完全用列表方法
    eqvec = list(np.zeros(n))
    bvec = []
    for i in range(n):
        eqvec[i] = orderingeq(n, i+1)
    # 原生列表的while判空可以正常执行
    while eqvec:
        # argmax可以兼容列表输入，返回最大值对应的索引
        index_max = int(np.argmax(eqvec))
        bvec.append(index_max)
        # 用pop按索引删除对应位置的元素，替代错误的remove逻辑
        eqvec.pop(index_max)
    return bvec

方案2：numpy原生高效实现 #

你整个循环的本质，就是把eqvec存储的数值按从大到小的顺序，提取每个值对应的原始索引。这个需求完全不需要反复循环找最大值、删元素，直接用numpy的排序接口就能一次性得到结果，尤其是n值较大时，性能远高于循环删除的写法：

import numpy as np

def ordering(n):
    eqvec = np.zeros(n)
    for i in range(n):
        eqvec[i] = orderingeq(n, i+1)
    # 对eqvec按降序排序，提取排序后的原始索引，就是要返回的bvec序列
    bvec = list(np.argsort(-eqvec))
    return bvec

注意：两种方案输出的结果完全等价——方案2去掉了循环中反复创建、删除数组元素的开销，是更符合numpy使用习惯的实现方式。

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