根据索引在 pandas 中制作多列

Question

我正在尝试将 pandas dataframe 列设置为索引。 由于它是一个分类列，我想折叠同一类别中的所有列。 例如：

df

  col1   Set_Index    col2
0    1        A         'a'
1    2        B         'b'
2    3        A         'c'
3    4        B         'd'

新的 df 必须如下所示：

编辑：我忘记将默认索引放在最终 output 所需的 output 的初稿中。 它必须最终看起来像：

new_df

             col1      col2
Set_Index 
   A      0    1       'a'
          2    3       'c'
   B      1    2       'b'
          3    4       'd'

我尝试使用 groupby() function 但没有多大帮助。

Answer 1

您可以将“Set_Index”列设置为索引，删除它并根据新索引对行进行排序。

df_new = df.set_index(df.Set_Index).drop(columns=['Set_Index']).sort_index()
df_new

您将拥有以下 df_new：

据我所知，每一行都应该有一个索引，我不知道如何创建您指定的 df_new（第 2 行和第 4 行没有索引）。

You can read more about indexing in pandas here: https://brohrer.github.io/dataframe_indexing.html