Iterate Python-Liste und die Summe gleich Elemente

Question

einen Endpunkt auf meinem Backend Unter Berücksichtigung, dass die folgende Antwort zurück:

class Arc_Edges_Data(Resource): 
    def get(self): 
     #Connect to databse 
     conn = connectDB() 
     cur = conn.cursor() 
     #Perform query and return JSON data 
     try: 
      cur.execute("select json_build_object('source', start_location, 'target', end_location, 'frequency', 1) from trips") 
     except: 
      print("Error executing select") 
     ArcList = list (i[0] for i in cur.fetchall()) 
     return ArcList 

Die Frequenz hier soll immer von 1 für jede Reise. Also diese ArcList stammt eine Antwort wie folgt aus:

[ 
    { 
     "frequency": 1, 
     "source": "c", 
     "target": "c" 
    }, 
    { 
     "frequency": 1, 
     "source": "a", 
     "target": "b" 
    }, { 
     "frequency": 1, 
     "source": "a", 
     "target": "b" 
    }, ... 
] 

Wie kann ich diese Reaktion durchlaufen und die Elemente zusammenzufassen, die die gleiche source und target haben? In diesem Fall hätte die resultierende Liste nur ein Paar Quelle/Ziel mit "a" und "b", aber die Häufigkeit wäre 2 wegen der Summe.

Ich weiß, dass für Javascript ich etwas wie Array.reduce verwenden könnte, aber ich glaube nicht, dass es für Python existiert.

Answer 1

Wie wäre es damit?

import collections 

data = [ 
    { 
     "frequency": 1, 
     "source": "c", 
     "target": "c", 
    }, 
    { 
     "frequency": 1, 
     "source": "a", 
     "target": "b", 
    }, 
    { 
     "frequency": 1, 
     "source": "a", 
     "target": "b", 
    }, 
] 

counter = collections.Counter() 

for datum in data: 
    counter[(datum['source'], datum['target'])] += datum['frequency'] 

print(counter) 

# Output: 
# Counter({('a', 'b'): 2, ('c', 'c'): 1}) 

Oh, wenn Sie die Daten wieder in das gleiche Format wieder setzen wollen, fügen Sie diesen Code:

newdata = [{ 
    'source': k[0], 
    'target': k[1], 
    'frequency': v, 
} for k, v in counter.items()] 

print(newdata) 

# Output: 
# [{'frequency': 1, 'target': 'c', 'source': 'c'}, {'frequency': 2, 'target': 'b', 'source': 'a'}] 

Answer 2

Sie können dies tun:

r = {} 
for d in ArcList: 
    key = (d['source'], d['target']) 
    r[key] = r.setdefault(key, 0) + d['frequency'] 
return [{'source': k[0], 'target': k[1], 'frequency': v} for k, v in r.items()] 

Wenn Sie möchten, bewahren Sie die ursprüngliche Reihenfolge der Artikel:

from collections import OrderedDict 
r = OrderedDict() 
# The rest of the solution is the same 
...