Python中循环处理向量至为空的代码异常问题求助

原代码存在的问题 #

• np.c_未正确调用（需通过numpy导入并使用全称），且缺少运算结果的存储逻辑
• 判断numpy数组是否为空的方式错误：OR==[]无法对numpy数组做有效判断，会返回布尔数组而非单个布尔值
• 循环逻辑顺序错误：初始取出的new未被处理，且当OR为空时仍会执行new=OR[0:3]，引发索引错误

修正后的代码（保持列向量输出） #

import numpy as np

OR = np.array([1,2,3,4,5,6])
# 初始化结果数组，保持列向量维度
result = np.array([]).reshape(0, 1)

while OR.size > 0:
    # 提取前3个元素
    new = OR[:3]
    # 执行运算，转为列向量后乘-2
    tran = -2 * np.c_[new]
    # 将运算结果追加到结果数组
    result = np.vstack([result, tran])
    # 删除原数组前3个元素
    OR = np.delete(OR, slice(0, 3))

print(result)

简化版本（一维数组输出） #

如果不需要列向量形式，可简化为：

import numpy as np

OR = np.array([1,2,3,4,5,6])
result = []

while OR.size > 0:
    new = OR[:3]
    tran = -2 * new
    result.extend(tran)
    OR = np.delete(OR, slice(0, 3))

# 转为numpy数组（可选）
result = np.array(result)
print(result)

关键修改点说明 #

1. 用OR.size > 0作为循环条件，准确判断numpy数组是否为空
2. 调整循环顺序：先处理当前提取的元素，再更新原始数组，避免初始元素遗漏
3. 新增结果存储逻辑，通过np.vstack（列向量）或列表extend（一维）保存运算结果
4. 使用slice(0,3)替代[0,1,2]，更简洁且适配数组切片操作

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