python - 将 3D 熊猫数据帧转换为 2d

Question

我有一个具有 2 级索引和一列的 3D 数据框，如下所示：

          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 个元素。我需要将其重塑为二维数据框，如下所示：

    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. 我该怎么做？

标签： pythonpandasdataframepivot

Answer 1

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

df_2d = df.unstack(1)

然后使用以下方法修复列级别：

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