Python

Question

我正在努力以一种简单有效的方式重组我的数据框。

第一个数据帧 'new' 是我所拥有的，所需的数据帧是 new2。我一直在摆弄 pivot、stack、unstack、set_index、reshape 等很长时间，仍然无法实现这种简单的转换。我不断收到错误消息或错误结果。有人可以帮忙吗？

我有什么

data1 = [50, 60]
data2 = [100, 200]
year = [2015,2016]
new = pd.DataFrame({'product': data1, 'year':year, 'market':data2})

我想要的

new2 = pd.DataFrame(columns = ['2015', '2016'])
new2.loc['market'] = data1
new2.loc['product'] = data2

desired outcome

Answer 1

你需要set_index with T and last for remove column name add rename_axis:

df = df.set_index('year').T.rename_axis(None, axis=1)
print (df)
         2015  2016
market    100   200
product    50    60

Python - 简单的数据框重组