从列表中删除最后一个元素

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 -loop：

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]