从列表中减去最后一个元素

Question

我需要减去最后一个元素，然后构建一个类似于[1,1,1,...]的向量。我有这个功能：

def vectores(lista):
    r=[]
    for e in lista:
        r.append(e[2])
        return r

在哪里

lista = [['pintor', 'NCMS000', 1], ['ser', 'VSIS3S0', 1], ['muralista', 'AQ0CS0', 1], ['diego_rivera', 'NP00000', 1], ['frida_kahlo', 'NP00000', 1], ['caso', 'NCMS000', 1]]

但是函数返回[1]；我能做什么？你知道吗

Answer 1

您将在循环的第一次迭代中返回。将return语句移到for-循环之外：

def vectores(lista):
    r=[]
    for e in lista:
        r.append(e[2])
    return r  # here

或者只使用列表：

def vectores(lista):
    return [e[2] for e in lista]