Разница между рядами в пандах

Question

У меня есть данные в CSV, которые я читаю с пандами. Данные в этом формате -

name    company income  saving  
A       AA       100     10

B       AA       200     20

Я хочу создать новую строку с именем A, компанией AA, а доход и сбережения - это разница A и B.

Ожидаемый результат-

name    company income  saving  
A       AA       -100     -10

Answer 1

Я считаю нужным:

print (df)
  name company  income  saving
0    A      AA     100      10
1    B      AA     200      20
2    C      AA     300      40

#for select columns by names
df1 = df[['name','company']].join(df[['income','saving']].diff(-1))
#for select columns by positions
#df1 = df.iloc[:, :2].join(df.iloc[:, 2:].diff(-1))

print (df1)
  name company  income  saving
0    A      AA  -100.0   -10.0
1    B      AA  -100.0   -20.0
2    C      AA     NaN     NaN