Как составить список списков

Question

Я python новичок, и у меня есть python DataFrame с 5 столбцами, как показано ниже. Мне нужно проверить, если значение в столбце a равно 1, а затем сложить соответствующее значение в столбце d и e вместе. Точно так же я бы проверил, равно ли значение в столбце a 2, затем сложил соответствующее значение в столбце d и e и т. Д. Значения в a = 3.

   a    b    c         d           e
   1    2   0.57735 -0.456086      9
   1    3   0.57735 -0.456086      9
   3    1   0.57735 -0.456086      9
   1    2   0.57735 -0.456086      9
   2    1   0.57735 -0.456086      9

Мой первоначальный подход выглядит следующим образом:

data= pd.read_csv('data.csv',usecols=['a','b','c','d','e'])
dis_j= pd.DataFrame(data)

if dis_j['at1']==1:
     sub_1 = np.stack(dis_j['d'], dis_j['e'])

Однако он не работает должным образом. Ждем ваших комментариев. Спасибо!