时间戳

首页 > 解决方案 > 熊猫重塑数据框，其中每行是单元格值和索引

python - 熊猫重塑数据框，其中每行是单元格值和索引

问题描述

如何获取正常的数据框，如下所示：

pd.DataFrame(columns=['X', 'Y'],index=['A','B'])

    X   Y
A   1   2
B   3   4

并生成一个新的数据框，其中每一行由单元格、列名和索引组成，如下所示：

       col1 col2 col3
row0    1     X    A  
row1    2     Y    A
row2    3     X    B
row3    4     Y    B

任何帮助表示赞赏。

标签： pythonpython-3.xpandasdataframe

解决方案

用于DataFrame.stack重塑，然后DataFrame.rename_axis用于新列名称，DataFrame.reset_index用于MultiIndex Seriesto DataFrame，如有必要，重命名索引并通过以下方式更改列的顺序DataFrame.reindex：

df = (df.stack()
        .rename_axis(('col3','col2'))
        .reset_index(name='col1')
        .rename(lambda x: f'row{x}')
        .reindex(['col1','col2','col3'], axis=1))
print (df)
      col1 col2 col3
row0     1    X    A
row1     2    Y    A
row2     3    X    B
row3     4    Y    B

Numpy/pandas 解决方案 - numpy.ravel, numpy.tile,numpy.repeat带有DataFrame构造函数：

df = pd.DataFrame({'col1': df.to_numpy().ravel(),
                   'col2': np.tile(df.columns,len(df.index)),
                   'col3': np.repeat(df.index,len(df.columns))}).rename(lambda x: f'row{x}')
print (df)
      col1 col2 col3
row0     1    X    A
row1     2    Y    A
row2     3    X    B
row3     4    Y    B

推荐阅读