Как суммировать элементы вложенного списка по условию?

Question

У меня есть вложенный список, подобный приведенному ниже, для которого я пытаюсь суммировать значения столбца 1, связанные с тем же значением в столбце 2, и добавить новый подсписок с суммой и еще один подсписок с «Всего ".

a = [
        ['45', '00128'], 
        ['88', '00128'], 
        ['87', '00128'], 
        ['50', '88292'], 
        ['69', '88292'], 
        ['70', '72415'], 
        ['93', '72415'], 
        ['79', '72415']
    ]

Мой текущий код выглядит так, как показано ниже, для которого, я полагаю, он будет работать с чем-то над списком a, но ничего не меняет внутри a.

for sl in a:
    x = sl[1]; c0=0
    if (sl[1] == x):
        c0 = c0 + int(sl[0])
    else:
        a.insert(a.index(sl)+1,[c0,''])
        a.insert(a.index(sl)+2,['Total',''])        

Результат, который я ищу, выглядит следующим образом:

b = [
        ['45',  '00128'], 
        ['88',  '00128'], 
        ['87',  '00128'],
        ['220', ''],      # This is 45 + 88 + 87
        ['Total', ''],
        ['50', '88292'], 
        ['69', '88292'], 
        ['119', ''],      # This is 50 + 69
        ['Total', ''],      
        ['70', '72415'], 
        ['93', '72415'], 
        ['79', '72415'],
        ['242', ''],      # This is 70 + 93 + 79
        ['Total', '']       
    ]

Как это сделать? спасибо

ОБНОВЛЕНИЕ

Входной список с 4 столбцами, как один слева, и необходимо суммировать col1, col3 и col4, чтобы получить правильное значение.

a = [                              >>  b = [
        ['45', '00128', '2','4'],  >>          ['45',    '00128', '2',     '4'    ], 
        ['88', '00128', '1','3'],  >>          ['88',    '00128', '1',     '3'    ], 
        ['87', '00128', '4','0'],  >>          ['87',    '00128', '4',     '0'    ], 
        ['50', '88292', '1','1'],  >>          ['220',   ''     , '7',     '7'    ],
        ['69', '88292', '9','5'],  >>          ['Total', '',      'Total', 'Total'],
        ['70', '72415', '8','9'],  >>          ['50',    '88292', '1',     '1'    ], 
        ['93', '72415', '3','2'],  >>          ['69',    '88292', '9',     '5'    ],
        ['79', '72415', '5','7']   >>          ['119',   '',      '10',    '6'    ],
    ]                              >>          ['Total', '',      'Total', 'Total'],
                                   >>          ['70',    '72415', '8',     '9'    ], 
                                   >>          ['93',    '72415', '3',     '2'    ], 
                                   >>          ['79',    '72415', '5',     '7'    ],
                                   >>          ['242',   '',      '16',    '18'   ],
                                   >>          ['Total', '',      'Total', 'Total'],
                                   >>       ]

Answer 1

Использование itertools.groupby

from itertools import groupby

result = []
for m,n in groupby(a, lambda x: x[1]):
    n = list(n)
    result.extend(n + [[sum(int(i) for i, _ in n), ""]])
print(result)

Вывод:

[['45', '00128'],
 ['88', '00128'],
 ['87', '00128'],
 [220, ''],
 ['50', '88292'],
 ['69', '88292'],
 [119, ''],
 ['70', '72415'],
 ['93', '72415'],
 ['79', '72415'],
 [242, '']]

Редактировать согласно комментарию

for m,n in groupby(a, lambda x: x[1]):
    n = list(n)
    val_1, val_2, val_3 = 0, 0, 0
    for i in n:
        val_1 += int(i[0])
        #val_2, val_3....
    result.extend(n + [[val_1, ""]])

Если вы можете использовать numpy, тогда сумма оси 0 проще

Пример:

for m,n in groupby(a, lambda x: x[1]):
    n = np.array(list(n), dtype=int)
    print(np.delete(np.sum(n, axis=0), 1))

np.delete --> Delete element in index 1
np.sum with axis=0 --> sum element in column. 

Answer 2

Довольно грубый способ сделать это. Это просто для ознакомления. Я бы порекомендовал ответ, данный Rake sh

from collections import defaultdict
    
sums_dict = defaultdict(int)
a = [
        ['45', '00128'],
        ['88', '00128'],
        ['87', '00128'],
        ['50', '88292'],
        ['69', '88292'],
        ['70', '72415'],
        ['93', '72415'],
        ['79', '72415']
    ]
sums_dict = {v:sums_dict[v]+int(k) for k,v in a}

for k in sums_dict:
    index = next((len(a) - i - 1 for i, lst in enumerate(reversed(a)) if k in lst), -1)
    a.insert(index, [str(sums_dict[k]),""])
    a.insert(index+2, ['Total', ''])

print(a)

Вывод:

[['45', '00128'], ['88', '00128'], ['87', ''], ['87', '00128'], ['Total', ''], ['50', '88292'], ['69', ''], ['69', '88292'], ['Total', ''], ['70', '72415'], ['93', '72415'], ['79', ''], ['79', '72415'], ['Total', '']]

Answer 3

Похоже, вы импортировали данные из электронной таблицы? Очень простой способ сделать это - использовать библиотеку pandas и позволить ей обрабатывать данные за вас:

a = [
        ['45', '00128'], 
        ['88', '00128'], 
        ['87', '00128'], 
        ['50', '88292'], 
        ['69', '88292'], 
        ['70', '72415'], 
        ['93', '72415'], 
        ['79', '72415']
]

# Import the library
import pandas as pd

# Put data into a pandas DataFrame and set column names
df = pd.DataFrame(a, columns=['value', 'category'])

# Change `value` column to integers
df['value'] = df['value'].astype(int)

# Group by the `category` column and sum
sum_df = df.groupby('category').sum()

# Show answer
print(sum_df)

Следует распечатать следующее:

          value
category
00128       220
72415       242
88292       119