python - 数据框转置 - 如何获得正确的列名称？

Question

我有这样的数据框:

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   

我想要这个数据框的形式像这样:

|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 解决方案(对我来说，重命名列并在转置后删除 ent 行是不优雅的)。

Answer 1

如果需要第一列是索引添加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