Python NumPy按条件展平数组 兼容空删除索引列表的实现方法

问题现状 #

现有代码支持传入二维坐标索引列表，将NumPy数组inv_r展平后删除对应索引的元素，但未处理索引列表为空的边界场景：

• 传入非空索引列表时，代码运行逻辑正常
• 传入空索引列表时，np.ravel_multi_index 直接抛出报错：ValueError: parameter multi_index must be a sequence of length 2
  需要兼容两类输入：
• 索引列表非空：保留原有按索引删除的逻辑
• 索引列表为空：直接返回展平后的完整数组，不执行删除操作

原始报错代码 #

import numpy as np

inv_r=(1e4)*np.array([[0.60800941, 0.79907128, 0.99442121],
       [0.61174008, 0.84891968, 0.71449188],
       [0.6211801 , 0.88869614, 0.91835812]])


List = []
Remove=np.ravel_multi_index(np.array(List).T, dims=inv_r.shape)
T1 = np.delete(inv_r.flatten(), Remove)
print([T1])

修复代码 #

报错原因是空输入下不需要执行多维索引转一维索引的逻辑，直接给np.delete传入空索引数组即可——np.delete原生支持空索引输入，此时不会删除任何元素，完全匹配空列表场景的需求，非空场景的原有逻辑完全保留。

注意：原代码用List作为变量名，和Python内置类型list重名，存在潜在冲突，修复时替换为语义明确的del_index_list。

import numpy as np

inv_r=(1e4)*np.array([[0.60800941, 0.79907128, 0.99442121],
       [0.61174008, 0.84891968, 0.71449188],
       [0.6211801 , 0.88869614, 0.91835812]])


del_index_list = []
if len(del_index_list) == 0:
    Remove = np.array([], dtype=np.intp)
else:
    Remove = np.ravel_multi_index(np.array(del_index_list).T, dims=inv_r.shape)
T1 = np.delete(inv_r.flatten(), Remove)
print([T1])

效果验证 #

• 传入空列表：代码无报错，返回长度为9的完整展平数组
• 传入非空索引（如[[0,0]]）：正常删除对应位置元素，返回长度为8的结果数组，和预期输出完全一致

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