如何删除NumPy二维数组中仅包含空元组的行

你的代码失效核心是matrix == ()的返回结果不符合预期：

• 纯整数数组用matrix == 0时，数值标量0会按NumPy广播规则逐元素和数组内的值比较，返回和原数组同形状的二维布尔数组，后续按axis=1做全True判断逻辑正常。
• 存储object类型的数组和空元组()做==运算时，NumPy不会触发逐元素比较的广播，只会返回一个维度为0的标量布尔值，后续调用np.all(..., axis=1)自然会抛出轴超出范围的错误。

两种经过验证的写法都能实现删除全空元组行的需求：

方法1：用np.vectorize封装逐元素判断 #

把“单个元素是否为空元组”的判断逻辑包装成可逐元素广播的函数，先生成形状匹配的布尔掩码，再做行过滤：

import numpy as np

matrix = np.array(
[[(1, 'foo'), (), (4, 'bar')],
 [(),(),()],
 [(1, 'foo'), (), (3, 'foobar')],
 [(),(),()]], 
dtype=object)

# 封装逐元素判断逻辑
check_empty = np.vectorize(lambda elem: elem == ())
# 过滤掉整行全为空元组的行
matrix = matrix[~np.all(check_empty(matrix), axis=1)]

运行后输出结果为：

array([[(1, 'foo'), (), (4, 'bar')],
       [(1, 'foo'), (), (3, 'foobar')]], dtype=object)

原数组中第2、4行全空元组的行被成功删除，符合预期。

方法2：用列表推导式逐行判断 #

小数据量场景下可以直接用Python原生逐行判断，避开NumPy对空元组的等值判断bug，逻辑更直观：

# 仅保留「不是所有元素都为空元组」的行
matrix = matrix[[not all(elem == () for elem in row) for row in matrix]]

运行结果和方法1完全一致，不需要额外调用NumPy的广播封装，代码可读性更高。

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