python - Pandas 将数据帧的行转换为对角线数据帧

Question

我有一个数据框，我想在其中将每一行转换为对角线数据框，并将所有生成的数据框绑定(bind)到 1 个大数据框。 输入:

            a   b   c
2021-11-06  1   2   3
2021-11-07  4   5   6

期望的输出:

                   a    b   c
Date                
2021-11-06     a    1   0   0
               b    0   2   0
               c    0   0   3
2021-11-07     a    4   0   0
               b    0   5   0
               c    0   0   6

我尝试在原始数据框的每一行上使用应用。

data = pd.DataFrame([[1,2,3],[4,5,6]], columns=['a', 'b', 'c'], index=pd.date_range('2021-11-06', '2021-11-07')) 

def convert_dataframe(ser):
    df_ser = pd.DataFrame(0.0, index=ser.index, columns=ser.index)
    np.fill_diagonal(df_ser.values, ser)
    return df_ser

data.apply(lambda x: convert_dataframe(x), axis=1)

但是，输出不是我预期的多索引数据帧。 输出是单个索引数据帧，其中每一行都是对返回的对角线数据帧的引用。

非常感谢任何帮助。提前致谢。

Answer 1

使用MultiIndex.droplevel在 DataFrame.stack 之后删除第一级 MultiIndex 和调用函数在 GroupBy.apply :

def convert_dataframe(ser):
    ser = ser.droplevel(0)
    df_ser = pd.DataFrame(0, index=ser.index, columns=ser.index)
    np.fill_diagonal(df_ser.values, ser)
    
    return df_ser



data = data.stack().groupby(level=0).apply(convert_dataframe)
print (data)
              a  b  c
2021-11-06 a  1  0  0
           b  0  2  0
           c  0  0  3
2021-11-07 a  4  0  0
           b  0  5  0
           c  0  0  6