我有一个这样的“sample.txt”。idx A B C D cat
J 1 2 3 1 x
K 4 5 6 2 x
L 7 8 9 3 y
M 1 2 3 4 y
N 4 5 6 5 z
O 7 8 9 6 z
有了这个数据集,我想得到行和列的和。
排成一排,没什么大不了的。
我做出了这样的结果。### MY CODE ###
import pandas as pd
df = pd.read_csv('sample.txt',sep="\t",index_col='idx')
df.info()
df2 = df.groupby('cat').sum()
print( df2 )
结果就是这样。A B C D
cat
x 5 7 9 3
y 8 10 12 7
z 11 13 15 11
但我不知道如何编写代码才能得到这样的结果。
(只需在A列和B列以及C列和D列中添加值)AB CD
J 3 4
K 9 8
L 15 12
M 3 7
N 9 11
O 15 15
有谁能帮忙写代码吗?
顺便说一下,我不想这样做。
(看起来太单调了,但如果这是唯一的办法,我会相信的)df2 = df['A'] + df['B']
df3 = df['C'] + df['D']
df = pd.DataFrame([df2,df3],index=['AB','CD']).transpose()
print( df )