如何将数组行中连续三个及以上重复值替换为0？代码死循环求助

死循环的直接原因 #

你代码里的column =+ 1是语法错误，这会把column赋值为1而非自增，导致循环条件column < column_nmbr永远成立，直接陷入死循环。正确写法应该是column += 1。但即使修正这个错误，原代码还存在多个逻辑缺陷：只处理第一行、仅替换单个元素、会出现索引越界（当column接近列数时，next_col2会超出数组范围）。

完整实现方案 #

下面提供两种可行的实现方式，都能完成“将每行连续三个及以上相同值替换为0”的需求：

方式一：逐行遍历标记 #

通过遍历每行元素，统计连续相同值的数量，标记需要替换的位置后统一置0：

import numpy as np

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

rows, cols = A.shape

for row_idx in range(rows):
    current_val = A[row_idx, 0]
    consecutive_count = 1
    replace_mask = np.zeros(cols, dtype=bool)
    
    for col_idx in range(1, cols):
        if A[row_idx, col_idx] == current_val:
            consecutive_count += 1
        else:
            # 标记连续数量≥3的区间
            if consecutive_count >= 3:
                replace_mask[col_idx - consecutive_count : col_idx] = True
            current_val = A[row_idx, col_idx]
            consecutive_count = 1
    # 处理行末尾的连续元素
    if consecutive_count >= 3:
        replace_mask[cols - consecutive_count : cols] = True
    
    # 执行替换
    A[row_idx, replace_mask] = 0

print(A)

方式二：用numpy卷积高效检测 #

利用一维卷积快速定位连续三个相同值的区间，适合处理大规模数组：

import numpy as np

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

rows, cols = A.shape
# 卷积核用于检测连续三个元素
kernel = np.array([1, 1, 1])

for row_idx in range(rows):
    row_data = A[row_idx]
    # 计算有效卷积结果
    conv_result = np.convolve(row_data, kernel, mode='valid')
    # 找到连续三个相同值的起始位置
    start_positions = np.where(conv_result == 3 * row_data[:-2])[0]
    # 标记所有需要替换的位置
    replace_mask = np.zeros(cols, dtype=bool)
    for pos in start_positions:
        replace_mask[pos:pos+3] = True
    # 执行替换
    A[row_idx, replace_mask] = 0

print(A)

最终输出 #

两种方式都会得到正确结果：

[[1 2 0 0 0 4]
 [1 3 2 4 2 4]
 [1 2 4 2 4 4]
 [1 2 3 0 0 0]
 [1 2 1 3 4 4]]

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