如何使用 NumPy 高效移除数组中间元素并用 None（或 NaN）填充末尾

心靈之曲 2026-01-03 00:00:00 次阅读

本文介绍在不使用 remove()、pop() 等内置列表方法的前提下，仅借助 numpy 和基础循环实现“逻辑删除”数组中指定值（如 2），并将空位统一补至末尾（填 none 或 np.nan）的两种专业方案。

在 NumPy 中，数组是固定长度的连续内存块，不支持原地“删除”或“收缩”——所谓“移除中间元素”，实质是保留非目标元素 + 补齐占位符。关键在于：避免低效 Python 循环，优先使用向量化操作；若需兼容 None，需注意 NumPy 数组类型限制（None 仅在 object 类型数组中合法，而数值运算更推荐 np.nan）。

✅ 推荐方案：纯 NumPy 向量化（高效、简洁）

import numpy as np

arr = np.array([1, 2, 3, 4, 5])

# 步骤1：构建布尔掩码，标记所有值为 2 的位置
mask = arr == 2

# 步骤2：用布尔索引提取非 2 元素，并转为 float（为容纳 np.nan 做准备）
filtered = arr[~mask].astype(float)

# 步骤3：在末尾补足 mask.sum() 个 np.nan
result = np.pad(filtered, (0, mask.sum()), constant_values=np.nan)

print(result)  # [ 1.  3.  4.  5. nan]

⚠️ 注意：np.pad() 的 (0, n) 表示“前端补 0 个，后端补 n 个”。constant_values=np.nan 是标准浮点缺失值标识，适用于科学计算；若业务强依赖 None，可改用 object 类型：result_obj = np.r_[arr[~mask].astype(object), np.full(mask.sum(), None)]

? 备选方案：基础循环（教学/兼容性场景）

原始代码的问题在于：

直接赋值 arr[i] = arr[i+1] 仅平移一位，未处理后续元素，且越界风险高；
arr[len(arr)-1] = None 报错是因为 int64 类型数组无法存 None。

修正后的安全循环实现（保持原数组长度，逻辑覆盖所有匹配项）：

def shift_with_none(arr, target=2):
    # 创建 object 类型数组以支持 None
    out = np.empty(len(arr), dtype=object)
    idx = 0  # 写入指针
    for x in arr:
        if x != target:
            out[idx] = x
            idx += 1
    # 剩余位置全部填 None
    out[idx:] = None
    return out

arr = np.array([1, 2, 3, 4, 5])
print(shift_with_none(arr))  # [1 3 4 5 None]