Python-重组数据框，将列名称移至行，调整数据框的形状

Question

我需要将df1转换为df2：

import pandas as pd
from pandas import DataFrame, Series

import numpy as np

df1 = pd.DataFrame(index=['date_1', 'date_2', 'date_3'], 
              columns=["A_count", "A_dollar", "B_count", "B_dollar"], 
              data=[[10,"$100",7,"$786"], [3,"$43",6,"$88"],     [5,"$565",8,"$876"]])
df1

基本上，我需要将项目（A和B）作为标签放在新列中，然后将第3列和第4列数据移动到A项下的每一行。 这将使我们在每个日期都有一个新行。

Answer 1

您可以通过以下划线将列转换为多索引，然后使用stack将其重整为长格式：

df1.columns = df1.columns.str.split("_", expand=True)
df1.stack(level=0).rename_axis((None, "item")).reset_index("item")

如果列名中有多个下划线，则如下所示：

df1 = pd.DataFrame(index=['date_1', 'date_2', 'date_3'], 
              columns=["A_x_count", "A_x_dollar", "B_y_count", "B_y_dollar"], 
              data=[[10,"$100",7,"$786"], [3,"$43",6,"$88"],     [5,"$565",8,"$876"]])
df1

您可以将rsplit与n = 1一起使用，以便仅在最后一个下划线处拆分：

df1.columns = df1.columns.str.rsplit("_", n=1, expand=True)
df1.stack(level=0).rename_axis((None, "item")).reset_index("item")