Haz un juego de 3 en raya con matrices

def tres_en_raya(matriz):
  for i in range(3):
    #Comprueba si hay tres iguales en la fila i
    if matriz[i] == [1, 1, 1] or matriz[i] == [2, 2, 2]:
      return True
    #Comprueba si hay tres iguales en la columna i
    if matriz[0][i] == matriz[1][i] and matriz[0][i] == matriz[2][i]:
        return True
    if matriz[0][0] == matriz[1][1] and matriz[0][0] == matriz[2][2]:
        return True
    if matriz[2][0] == matriz[1][1] and matriz[2][0] == matriz[0][2]:
        return True
  return False

matriz1 = [
    [1, 1, 1],
    [0, 0, 0],
    [0, 0, 0]
]