我有一个3D数据框,其中包含2个级别的索引和一个看起来像这样的列:
col1
0 0 67.23
0 1 7382
0 2 43
.
.
0 8002 54
0 8003 87
1 0 348
1 1 83
1 2 234
.
.
1 8002 23
1 8003 87
....
9 0 348
9 1 833
9 2 43433
.
.
9 8002 23
9 8003 87
第一级具有10个索引,第二级每个具有8004个元素。我需要将其重塑为2d数据框,如下所示:
0 | 1 | 2 | 3 | ... | 8000 | 8001 | 8002 | 8003
--------------------------------------------------
0 | | | | | | | |
1 | | | | | | | |
2 | | | | | | | |
. | | | | | | | |
. | | | | | | | |
8 | | | | | | | |
9 | | | | | | | |
为此,我创建了一个新索引:new_idx = pd.Index(range(0,10))
,新列为:cols = range(0,8004)
,然后尝试按以下方式使用pivot
:
2d_df = df.pivot(index=new_idx, columns=cols, values='var1')
但这给我一个错误:Int64Index([0,1,2,3,4,5,6,7,8,9], dtype='int64') not in index
。我应该怎么做?
答案 0 :(得分:1)
尝试使用DataFrame.unstack
-请记住,如果索引重复,这将不起作用。
df_2d = df.unstack(1)
然后使用以下命令修复列级别:
df_2d.columns = df_2d.columns.droplevel(0)