如何快速筛选 NumPy 数组中仅包含整数的元素？

本文共计753个文字，预计阅读时间需要4分钟。

在处理包含无法避免生成混合类型（包含+ndarray和标签整数）的numpy object数组时，高效筛选出其中所有子数组、剔除纯Python整数或numpy标签的方法包括：

当 NumPy 数组的 dtype=object 且元素类型不一致（例如同时包含 np.ndarray 和 Python int 或 np.int64）时，NumPy 的向量化操作将失效——因为底层无法对异构对象执行统一的 dtype 检查或逻辑运算。此时看似“避免循环”的诉求，实则受限于 Python 对象模型的本质：object 数组本质上是带 NumPy 外壳的 Python 列表，任何真正的类型判断都必须逐元素调用 isinstance()。

✅ 推荐方案（简洁、清晰、实际最快）：

import numpy as np mixed_array = np.array([np.array([1, 2]), np.array([1, 2]), 0], dtype=object) # 使用列表推导式 + isinstance —— 简洁且通常比 vectorize 更快 filtered = np.array([x for x in mixed_array if isinstance(x, np.ndarray)]) print(filtered) # 输出: [array([1, 2]) array([1, 2])]

⚠️ 注意：isinstance(x, np.ndarray) 同时兼容 np.ndarray 实例和 np.matrix（已弃用），但不匹配 NumPy 标量（如 np.int32(0)、np.float64(1.0)）。若需同时保留标量数组（如 np.array(0)）和多维数组，应改用：

# 匹配所有“数组状”对象（含 0 维数组） filtered = np.array([x for x in mixed_array if np.ndim(x) >= 0 and hasattr(x, '__len__') or isinstance(x, np.ndarray)]) # 更稳妥写法（推荐）： filtered = np.array([x for x in mixed_array if hasattr(x, 'shape') and not np.isscalar(x)])

❌ 不推荐 np.vectorize 方案：

isarray = np.vectorize(lambda x: isinstance(x, np.ndarray)) filtered = mixed_array[isarray(mixed_array)] # 语法可行，但内部仍隐式循环，且额外开销更大

np.vectorize 仅为语法糖，不提升性能，反而引入函数封装与返回值处理的额外成本，在大数据量下显著慢于原生列表推导。

? 进阶需求：将整数转为单元素数组（如 0 → np.array([0])）
若业务逻辑允许统一升维，可一步完成类型归一化：

normalized = np.array([ x if isinstance(x, np.ndarray) else np.atleast_1d(x) for x in mixed_array ]) # 结果: [array([1, 2]) array([1, 2]) array([0])]

np.atleast_1d() 安全处理标量、0维数组及已有数组，是比 np.array([x]) 更鲁棒的选择（后者对 np.array([1,2]) 会嵌套成 array([array([1, 2])])）。

? 总结：

• Object 数组是性能瓶颈的根源，最优解永远是重构上游逻辑，避免生成混合类型数组；
• 若必须处理，列表推导式 + isinstance(x, np.ndarray) 是最直接、可读且实际最快的方案；
• 切勿迷信 vectorize 或 dtype 属性过滤（mixed_array.dtype == object 无法区分内部元素类型）；
• 对整数补全为数组的需求，优先使用 np.atleast_1d 而非手动包装。