的Python：列表添加

我有兩個表像below.i是從數據庫的Python：列表添加

EmpID = Assign.objects.select_related().filter(pName=selProject).filter() 
    .order_by('laEmpNum').values_list('laEmpNum', flat=True) 

TotDur = Assign.objects.select_related().filter(pName=selProject).order_by('laEmpNum') 
    .values_list('duration', flat=True) 

EmpID = [u'1046', u'1046', u'1046', u'8008', u'8008', u'8011'] 

TotDur = [0.0, 2.0, 2.5, 0.0, 2.7, 1.2]

得到這個如果EmpIDs在TotDur相同，則相應的值應該收集並添加（總和）。

ResOne = 0.0 + 2.0 + 2.5 i.e 4.5 
ResTwo = 0.0+2.7   i.e 2.7 
ResThr = 1.2    i.e 1.2

如何在Python中執行此操作。

來源

2012-11-17 molakalapalli ramesh

順序如果這些元素均使用一個OrderedDict在你在你的例子已經顯示順序，可以使用itertools.groupby

from itertools import groupby 
from operator import itemgetter 
[(k , sum(e for _,e in v)) for k,v in groupby(zip(EmpID, TotDur), itemgetter(0))] 
[(u'1046', 4.5), (u'8008', 2.7), (u'8011', 1.2)]

INFACT你不需要創建兩個單獨的列表，後來壓縮它

Emp_TotDur = Assign.objects.select_related().filter(pName=selProject).filter() 
    .order_by('laEmpNum').values_list('laEmpNum', 'duration') 

[(k , sum(e for _,e in v)) for k,v in groupby(Emp_TotDur, itemgetter(0))]

來源

2012-11-17 09:42:24 Abhijit

您可以使用defaultdict：

In [60]: from collections import * 

In [61]: EmpID = [u'1046', u'1046', u'1046', u'8008', u'8008', u'8011'] 

In [62]: TotDur = [0.0, 2.0, 2.5, 0.0, 2.7, 1.2] 

In [63]: d=defaultdict(int) 

In [64]: for x,y in zip(EmpID,TotDur): 
    d[x]+=y 
    ....:  

In [65]: d 
Out[65]: defaultdict(<type 'int'>, {u'8008': 2.7, u'1046': 4.5, u'8011': 1.2})

或者乾脆dict：

In [70]: d=dict() 

In [71]: for x,y in zip(EmpID,TotDur): 
    d[x]=d.get(x,0)+y 
    ....:  

In [72]: d 
Out[72]: {u'1046': 4.5, u'8008': 2.7, u'8011': 1.2}

來源

2012-11-17 09:37:25

defaultdict是一個很好的數據結構來使用，爲int領域它假定新的密鑰值0，並允許方便收集桶桶：

from collections import defaultdict 
d = defaultdict(int) 
for i, j in zip(EmpID, TotDur): 
    d[i] += j 
print d # defaultdict(<type 'int'>, {u'8008': 2.7, u'1046': 4.5, u'8011': 1.2})

來源

2012-11-17 09:39:33

如何從我的答案有何不同？ –

它不是，我沒有注意到你的，直到我發佈我的;） –

@ AshwiniChaudhary：爲什麼它需要有所不同？它確實解釋了「int」行爲，我發現它很有用。 –

你能做到這樣

l1 = [1,1,2,3,3] 
l2 = [1,2,3,4,5] 
last_val=l1[0] 
sum=0 
list=[] 
for pair in zip(l1,l2): 
    if pair[0]!=last_val: 
     print(sum) 
     list.append(sum) 
     sum=0 
    last_val=pair[0] 
    sum+=pair[1] 
list.append(sum) 
print(list)

來源

2012-11-17 09:39:34

可以zip兩個列表，其中對第一至第二的元素融入到元組的元素。然後遍歷它們，並使用元組的第一部分作爲鍵在詞典：

EmpID = [u'1046', u'1046', u'1046', u'8008', u'8008', u'8011'] 
TotDur = [0.0, 2.0, 2.5, 0.0, 2.7, 1.2] 
Result = {} 

for (key, value) in zip(EmpID, TotDur): 
    if not key in Result: 
     Result[key] = value 
    else: 
     Result[key] += value 

# Print the result 
for (key, value) in Result.items(): 
    print key, value

你可能想，如果你想保留的EMPS

來源

2012-11-17 09:40:47

試試這個，它將所有對應的id值的總和放在一個列表中，稍後可以用empId訪問列表並查看它的總和。

finalList = [] 
lastEmpId 

for index, emp in enumarate(EmdIP): 
    if lastEmpId == emp: 
     finalList[lastEmpId] += TotDur[index] 
    else: 
     finalList[lastEmpId] = TotDur[index] 
    lastEmpId = emp; 

print finalList

來源

2012-11-17 09:44:24

python 3.2 
    res=[] 
    for i in set(ID): 
     b=[] 
     for x,y in enumerate(ID): 
       if i==y: b.append(T[x]) 
    res.append(sum(b))

來源

2012-11-17 18:15:03 raton

的Python：列表添加

回答

相關問題