我有这样的数据框:
import pandas as pd
df = pd.DataFrame({'a': [1, 2], 'b': [3, 4], 'ent':['A','B']})
| |a |b |ent
|---|---|----|----
|0 |1 |3 |A
|1 |2 |4 |B
我希望这个数据框的形式如 this:
|ent |A |B
-------------
|a |1 |2
|b |3 |4
我可以申请df.T
或df.transpose
但我无法获得我想要的内容(标题):
| |0 |1
-------------
|a |1 |2
|b |3 |4
|ent |A |B
我正在寻找pandastic解决方案(对我来说,在转置后重命名列和删除行是不优雅的。)
答案 0 :(得分:2)
如果需要第一列index
添加set_index
:
df1 = df.set_index('ent').T
print (df1)
ent A B
a 1 2
b 3 4
如果需要默认索引:
df1 = df.set_index('ent').T.rename_axis(None, 1).rename_axis('ent').reset_index()
print (df1)
ent A B
0 a 1 2
1 b 3 4
答案 1 :(得分:1)
pd.pivot_table(df, values=['a','b'], columns='ent')
完全符合您的要求
编辑:jezrael是完全正确的。使用
显示的pivot_table函数的输出 df= a b ent
0 1 3 A
1 2 4 B
是
ent A B
a 1 2
b 3 4
但如果我们将df更改为
df= a b ent
0 1 3 A
1 2 4 A
pd.pivot_table(df, values=['a','b'], columns='ent')
的输出是
ent A
a 1.5
b 3.5
。
请注意,可以在pivot_table参数aggfunc中应用自己的函数。 '平均'只是默认值。