Pandas中数据框的两行标题到一个行标题

Question

我有一个带有多索引、二维标题的数据集。我想通过在同一行中删除第一行中的空列名来将其合并到一个标题行中。

下面显示了我拥有的数据框的结构。 前两行是标题。

         id  One             Two      
  response   X1   Y1   Z1    X2    Y2
0        0  1.1  1.2  1.4  1.11  1.22
1        1  1.1  1.2  1.3  1.11  1.22
2        2  1.1  1.2  1.1  1.11  1.22

          

我想把上面的数据框改成下面的一个，

   id  One 1.X1  One 2.Y1  One 3.Z1  Two 1.X2  Two 2.Y2
0   0       1.1       1.2       1.4      1.11      1.22
1   1       1.1       1.2       1.3      1.11      1.22
2   2       1.1       1.2       1.1      1.11      1.22

实际数据框有 100 多列。

希望有人可以在这里帮助我。

比你好多了。

玛丽·阿宾。

Answer 1

如果您的列确实是MultiIndex

即

print(df.columns)

MultiIndex([( 'id', 'response'),
            ('One',       'X1'),
            ('One',       'Y1'),
            ('One',       'Z1'),
            ('Two',       'X2'),
            ('Two',       'Y2')],
           )

然后我们可以将它们传递到一个新的数据框中，并在展平列之前在第一级使用累积计数。

s = pd.DataFrame.from_records(df.columns)
s['col'] =  (s.groupby(0).cumcount()+1).astype(str) + '.'
#skip the first row and re-order columns to match your desired order.
df.columns = ['id'] + s.iloc[1:, [0,2,1]].astype(str).agg(' '.join,1).tolist()

---

print(df)

   id  One 1. X1  One 2. Y1  One 3. Z1  Two 1. X2  Two 2. Y2
0   0        1.1        1.2        1.4       1.11       1.22
1   1        1.1        1.2        1.3       1.11       1.22
2   2        1.1        1.2        1.1       1.11       1.22


print(s)
    0         1 col
0   id  response  1.
1  One        X1  1.
2  One        Y1  2.
3  One        Z1  3.
4  Two        X2  1.
5  Two        Y2  2.

Answer 2

df.columns = df.columns.droplevel(0)

查看https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.MultiIndex.droplevel.html