数据帧转置 - 如何使列名称正确?

时间:2018-03-22 11:22:48

标签: python pandas dataframe

我有这样的数据框:

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.Tdf.transpose但我无法获得我想要的内容(标题):

|     |0  |1
-------------
|a    |1  |2
|b    |3  |4
|ent  |A  |B

我正在寻找pandastic解决方案(对我来说,在转置后重命名列和删除行是不优雅的。)

2 个答案:

答案 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中应用自己的函数。 '平均'只是默认值。