Как добавить несколько столбцов в фрейм данных python с помощью других столбцов фрейма данных

Question

Вот мое требование

У меня есть существующий фрейм данных df.A[a,b,c], и я хотел бы создать новый df.B[X,Y] из df.A, выполнив некоторую арифметическую операцию c над столбцами в df.A

это будет похоже на

df.A= a b c
      0 1 2
      0 2 0
      1 3 2

My df.B будет получено как

df.B['X','Y']=df.A[(sum[A.a]+[A.b]),sum[A.c]]

Результат должен выглядеть как

df.B= X Y
      7 4

дайте мне знать, если вам понадобятся какие-либо дополнительные сведения для решения этого дела.

Answer 1

Ваша единственная цель - получить сумму (A + B) и сумму (C)?

Если это так, просто сделайте что-то вроде этого:

import pandas as pd
import numpy as np

df1 = pd.DataFrame({'A':np.random.randint(0,10,3),
                    'B':np.random.randint(0,10,3),
                    'C':np.random.randint(0,10,3)})
df1.groupby(df1.index.isin(['A','B'])).sum().rename({False: 'Y', True: 'X'})

Мне трудно понять цель этого, поэтому, если вы добавите больше контекста, это поможет создать более элегантное решение.

Думаю, было бы проще просто получить выходы:

A_plus_B = df3[['A','B']].to_numpy().sum()
C = df3['C'].sum()