二维数组中的前N个值具有要掩码的重复项

def mask_duplicates(array, N):
    flattened_array = [element for sublist in array for element in sublist]
    count = {}
    masked_array = []
    for element in flattened_array:
        count[element] = count.get(element, 0) + 1
        if count[element] <= N:
            masked_array.append(element)
        else:
            masked_array.append(None)  # 或者使用标志位表示被掩码
    return [masked_array[i:i+len(array[0])] for i in range(0, len(masked_array), len(array[0]))]