jedná se o cvičení v pythonu, nemohu použít pandy jen vestavěné funkce pythonu. Mám tento seznam:
['a', 1],
['b', 1],
['a', 2],
['a', 2],
Výstup by měl vypadat takto:
['a', 5],
['b', 1],
Toto je kód jsem tak daleko, ale já jsem stále v podstatě stejný seznam.
result = []
for x,y in list_data:
if x not in result:
result.append([x, int(y)])
else:
result[1] += int(y)
result
Můžete si vytvořit slovník vypočítat součty a pak převést zpět na seznam:
data = [['c', 2], ['a', 1], ['b', 1], ['a', 2], ['a', 2], ['c', 3]]
totals = {} # could use collections.defaultdict for this part
for k, v in data:
totals[k] = totals.get(k, 0) + v
print(totals) # {'c': 5, 'a': 5, 'b': 1}
output = [[k, v] for k, v in totals.items()]
# or, output = list(map(list, totals.items()))
# or, output = list(totals.items()) # if you are fine with a list of tuples
print(output) # [['c', 5], ['a', 5], ['b', 1]]
# if you want the output to be sorted alphabetically
output = sorted(output, key=lambda lst: lst[0])
print(output) # [['a', 5], ['b', 1], ['c', 5]]
c ve vstupu. One-liner:
import itertools
[[x[0], sum([value[1] for value in x[1]])] for x in itertools.groupby(sorted(list_data), lambda i: i[0])]
Podrobnosti:
[[x[0], sum([value[1] for value in x[1]])] for x in itertools.groupby(sorted(list_data), lambda i: i[0])]
sorted(list_data) # sort the list alphabetically
itertools.groupby(sorted(list_data), lambda i: i[0]) # group by the "key"
# result: [('a', <itertools._grouper object at 0x000001D6B806F700>), ('b', <itertools._grouper object at 0x000001D6B806F310>)]
x[0] for x in # ['a', 'b']
[value[1] for value in x[1]] for x in # [[1, 2, 2], [1]]
sum([value[1] for value in x[1]]) for x in # [5, 1]
x[0], sum([value[1] for value in x[1]]) for x in # [['a', 5], ['b', 1]]
Slovník je nejlepší praxe pro váš úkol.
data = [['a', 1], ['b', 1], ['a', 2], ['a', 2]]
d = {}
for key, value in data:
d[key] = value + (d[key] if key in d else 0)
# generate list of tuples and sort
out = sorted(d.items())
print(out)
Zdá se, že jste přemýšlel, seznamy jako slovník. Pro váš kód do práce, budete muset najít výsledek index, který chcete přidat hodnotu (výsledek[1]).
Ale pokud jste změnit výsledek do slovníku můžete mít téměř stejný kód:
result = dict() # or result = {}
for x,y in list_data:
if x not in result:
result[x] = int(y)
else:
result[x] += y
print(result)
Out[1]: {'a': 5, 'b': 1}
Pokud chcete mít seznam, můžete použít:
result_list = []
for x,y in result.items():
result_list.append([x,y])
print(result_list)
[['a', 5], ['b', 1]]