Pandasでひとつ前の列を加算したい

Question

Pandasでひとつ前の列を積上げした結果にしたいのですがどのように計算すればいいのでしょうか

元データ

	1	2	3
A	3	3	1
B	0	3	1
C	1	0	1
D	1	0	1

結果
1：元1
2：元1+元2
3：元1+元2+元3

	1	2	3
A	3	6	7
B	0	3	4
C	1	1	2
D	1	1	2

import io
import pandas as pd


data = """
A,3,3,1
B,0,3,1
C,1,0,1
D,1,0,1
"""
df = pd.read_csv(io.StringIO(data), header=None, index_col=0)

Answer 1

これでどうでしょう

import io
import pandas as pd

data = """
A,3,3,1
B,0,3,1
C,1,0,1
D,1,0,1
"""
df = pd.read_csv(io.StringIO(data), header=None, index_col=0)

df.iloc[:, 2] = df.iloc[:, 0:3].sum(axis=1)
df.iloc[:, 1] = df.iloc[:, 0:2].sum(axis=1)
#   1  2  3
#0
#A  3  6  7
#B  0  3  4
#C  1  1  2
#D  1  1  2

これ以上列数が長かったらループかな。。

---

【追記】

よくよく考えたら、もう少しスマートな方法があった・・・

import io
import pandas as pd

data = """
A,3,3,1
B,0,3,1
C,1,0,1
D,1,0,1
"""
df = pd.read_csv(io.StringIO(data), header=None, index_col=0)

df = df.apply(lambda d: d.cumsum(), axis=1)
#   1  2  3
#0
#A  3  6  7
#B  0  3  4
#C  1  1  2
#D  1  1  2

Comments

[barobaro]

cumsumはじめてみました。
最初のプログラムも勉強になります。
どうもありがとうございました