python比较列表的子列表的值并添加值

时间:2013-06-15 21:25:33

标签: python list

我有一个包含未知数量的子列表的列表。 我想总是将一个子列表与 ALL 其他子列表进行比较。 如果位置0,3和5的第一个子列表的值等于任何其他子列表,我想在所有匹配列表中将位置7的值加在一起。 然后将第一个子列表与第7位的(新添加的值)添加到新列表中。

list = [['A', 'a', 'b', 'B', 'c', 'd', 'C', '1'], 
         ['A', 'a', 'b', 'B', 'c', 'd', 'C', '3'], 
         ['D', 'r', 's', 'E', 't', 'u', 'F', '2'], 
         ['A', 'a', 'b', 'B', 'c', 'd', 'C', '2'], 
         ['D', 'r', 's', 'E', 't', 'u', 'F', '2'],.....]

想要输出:

new_list = [['A', 'a', 'b', 'B', 'c', 'd', 'C', '6'], 
             ['D', 'r', 's', 'E', 't', 'u', 'F', 4],...]

我写了这段代码

def Inter(list):
a = 0
b = 3 
c = 5 
d = 0
x = [] 

for i in range(len(list)): 
    for y in range(len(list)):
        if list[i][a] == list[y][a] and list[i][b] == list[y][b] and list[i][c] == list[y][c]: 
            IntSumtemp = []
            IntSumtemp.append(str(float(list[i][7]) + float(list[y][7])))
            x.append(list[i] + IntSumtemp) 
            del (x[d][7])
            d +=1
        else: None
return x
new_list= Inter(list)

但它给出了这个输出:

new_list= [['A', 'a', 'b', 'B', 'c', 'd', 'C', '2.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '4.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '3.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '4.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '6.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '5.0'], 
       ['D', 'r', 's', 'E', 't', 'u', 'F', '4.0'], 
       ['D', 'r', 's', 'E', 't', 'u', 'F', '4.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '3.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '5.0'], 
       ['A', 'a', 'b', 'B', 'c', 'd', 'C', '4.0'], 
       ['D', 'r', 's', 'E', 't', 'u', 'F', '4.0'],
       ['D', 'r', 's', 'E', 't', 'u', 'F', '4.0']]

有人可以帮帮我吗? (对不起,我是一个绝对的初学者,所以如果有什么不清楚,请问......)

1 个答案:

答案 0 :(得分:1)

这种方法是O(N),而不是你的O(N ^ 2)

from collections import OrderedDict
from operator import itemgetter

items = [['A', 'a', 'b', 'B', 'c', 'd', 'C', '1'], 
         ['A', 'a', 'b', 'B', 'c', 'd', 'C', '3'], 
         ['D', 'r', 's', 'E', 't', 'u', 'F', '2'], 
         ['A', 'a', 'b', 'B', 'c', 'd', 'C', '2'], 
         ['D', 'r', 's', 'E', 't', 'u', 'F', '2']]


key = itemgetter(0, 3, 5)
d = OrderedDict()

for x in items:
    d.setdefault(key(x), x[:7] + [0])[7] += int(x[7])

print d.values()

[['A', 'a', 'b', 'B', 'c', 'd', 'C', 6], ['D', 'r', 's', 'E', 't', 'u', 'F', 4]]