Найти в матрице заданный фрагмент

from random import randint
 
b = [[0, 1, 1], [1, 0, 1], [1, 1, 0]]  # заданный фрагмент
 
x, y = 6, 6  # размерность большой матрицы
a = [[randint(0, 1) for j in range(x)] for i in range(y)]  # создание и заполнение большой матрицы
for i in range(len(a)):
    for j in range(len(a[i])):
        print(a[i][j], end=' ')
    print("")
print("\n\n")
 
for i in range(len(b)):  # Вывод заданного фрагмента для наглядности
    for j in range(len(b[i])):
        print(b[i][j], end=' ')
    print()
 
flag = 0
kol = 0
for i in range(len(a)):
    for j in range(len(a[i])):
        temp = []
        if (i + 1 + len(b)) < len(a) and (j + 1 + len(b[i])) < len(a[i]):
            for n in range(len(b)):
                temp.append([])
                for m in range(len(b[n])):
                    temp[n].append(a[i + n][j + m])
        else:
            break
        for n in range(len(b)):
            for m in range(len(b[n])):
                if temp[n][m] != b[n][m]:
                    flag = 1
        if flag == 0:
            kol+=1
print(kol)

from random import randint
 
b = [[0, 1], [1, 0], [1, 1]]  # заданный фрагмент
 
a = [[randint(0, 1) for _ in range(10)] for _ in range(10)]
 
print(*b, '-', *['\t'.join([str(j) for j in i]) for i in a], sep='\n')
 
res = []
 
for i in range(len(a) - len(b) + 1):
    for j in range(len(a[0]) - len(b[0]) + 1):
        if a[i][j:j+len(b[0])] == b[0]:
            res.append((i, j))
            for x in range(len(b)):
                if a[i+x][j:j+len(b[0])] != b[x]:
                    del res[-1]
                    break
 
if res:
    for i in res:
        a[i[0]][i[1]] = '\x1b[6;30;42m' + str(a[i[0]][i[1]]) + '\x1b[0m'
    print('\nнайденные элементы: ', *res, *['\t'.join([str(j) for j in i]) for i in a], sep='\n')
else:
    print('элементов не найдено')