将3D熊猫数据框转换为2d

时间:2018-12-17 13:01:41

标签: python pandas dataframe pivot

我有一个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。我应该怎么做?

1 个答案:

答案 0 :(得分:1)

尝试使用DataFrame.unstack-请记住,如果索引重复,这将不起作用。

df_2d = df.unstack(1)

然后使用以下命令修复列级别:

  df_2d.columns = df_2d.columns.droplevel(0)