根據索引在 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