Bebeu mais menos

# coding: utf-8
# Quem bebeu mais menos
# Amanda Luna [116210896] / UFCG

def quem_bebeu_mais_menos(sab, dom) :

    lista1 = []
    lista2 = []
    lista3 = []
    soma1 = 0
    soma2 = 0
    soma3 = 0

    for i in range(len(sab)) :

        soma1 = coluna(sab,i)
        lista1.append(soma1)

    for i in range(len(dom)) :

        soma2 = coluna(dom,i)
        lista2.append(soma2)

    for i in range(len(lista1)) :

        soma3 = lista1[i] + lista2[i]
        lista3.append(soma3)
        soma3 = 0

    tup = maior_menor(lista3)

    return tup

def coluna(matriz,ind) :

    soma = 0

    for i in range(len(matriz)) :

        soma += matriz[i][ind]

    return soma

def maior_menor(lista):

    maior = lista[0]
    menor = lista[0]
    a = 0
    b = 0


    for i in range(len(lista)) :

        if lista[i] > maior :
            maior = lista[i]
            a = i

        if lista[i] < menor :
            menor = lista[i]
            b = i

    a += 1
    b += 1

    tup = (a,b)

    return tup

sabado = [[1,2,3], [0,1,0], [3,1,2]]
domingo = [[2,1,3], [0,2,1], [1,1,2]]
assert quem_bebeu_mais_menos(sabado, domingo) == (3,1)

sabado = [[1,2,3,4,5], [0,1,2,3,1], [2,1,0,1,2], [3,1,2,1,3], [4,1,3,0,0]]
domingo = [[0,1,1,0,1], [1,2,2,0,2], [2,3,1,1,1], [3,4,2,0,0], [4,3,3,0,0]]
assert quem_bebeu_mais_menos(sabado, domingo) == (1,4)