Question

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

Answer 1

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

df_2d = df.unstack(1)

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

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

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

1 个答案: