quantidade de colunas python code example

Example: quantidade de colunas python

def obter_dimensao(m):
    # Verifica se todas as linhas da matriz
    # possuem o mesmo tamnho
    if len({len(i) for i in m}) > 1:
        raise TypeError('Matriz 2D invalida.')

    # Calcula quantidade de linhas na matriz
    linhas = len(m)

    # Se nao houverem linhas na matriz
    # assume zero colunas
    colunas = len(m[0]) if linhas else 0

    return (linhas, colunas)

m1 = []                  # 0x0
m2 = [[],[],[]]          # 3x0
m3 = [[1],[2],[3]]       # 3x1
m4 = [[1,2,3],[1,2,3]]   # 2x3
m5 = [[1,2,3],[1,2,3,4]] # Matriz invalida!

print(obter_dimensao(m1))
print(obter_dimensao(m2))
print(obter_dimensao(m3))
print(obter_dimensao(m4))
print(obter_dimensao(m5))