verificar si una matriz es cuadrada

Verificar si una matriz es cuadrada

Enunciado: Escribe una función que determine si una matriz es cuadrada (mismo número de filas y columnas).

Programa

def es_cuadrada(matriz):
    filas = len(matriz)
    for fila in matriz:
        if len(fila) != filas:
            return False
    return True

# Ejemplo
M1 = [[1, 2], [3, 4]]
M2 = [[1, 2, 3], [4, 5, 6]]
print(es_cuadrada(M1))  # True
print(es_cuadrada(M2))  # False