如何检查数组行的特定比特序列并生成对应4比特数组？

问题根源 #

1. 数组匹配逻辑错误：原代码用any([...] == fb[i])判断行匹配，any()只要有一个元素相等就返回True，不符合“整行完全匹配”的需求，且直接数组比较返回布尔数组，容易触发真值歧义错误。
2. 运算符误用：用位运算&替代逻辑与and，&会返回布尔数组，而if判断需要单个布尔值，导致报错。
3. 索引越界：5比特行的最后一个元素索引是4（数组索引从0开始），原代码写fb[i-1,5]会触发索引错误。
4. 边界未处理：当i=0时，i-1=-1会访问数组最后一行，不符合“前一行”的逻辑，需单独处理。

修改后的代码 #

import numpy as np

array = []
for i in range(len(fb)):
    # 匹配全1的5比特行
    if np.array_equal(fb[i], [1, 1, 1, 1, 1]):
        array.append([9, 9, 9, 9])
    # 匹配[0,0,1,0,0]行
    elif np.array_equal(fb[i], [0, 0, 1, 0, 0]):
        array.append([1, 0, 1, 0])
    # 仅当i>0（存在前一行）时，匹配[0,1,0,1,1]行
    elif i > 0 and np.array_equal(fb[i], [0, 1, 0, 1, 1]):
        # 取前一行最后一个元素（索引4）判断
        if fb[i-1, 4] == 0:
            array.append([0, 1, 1, 0])
        else:
            array.append([0, 0, 0, 0])
    # 可根据需求添加其他情况的默认处理
    else:
        array.append([0, 0, 0, 0])

关键修改说明 #

• 用np.array_equal()判断整行是否完全匹配，返回单个布尔值，彻底解决真值歧义问题。
• 用逻辑与and替代位运算&，确保条件判断返回单个布尔值。
• 修正索引为4，避免越界错误。
• 添加i>0的判断，处理首行无前置行的边界场景。

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