将熊猫的宽形状重塑为长
- 2024-12-27 08:46:00
- admin 原创
- 115
问题描述:
假设我在 Pandas 中有以下数据框:
AA BB CC
date
05/03 1 2 3
06/03 4 5 6
07/03 7 8 9
08/03 5 7 1
我想将其转换为以下内容:
AA 05/03 1
AA 06/03 4
AA 07/03 7
AA 08/03 5
BB 05/03 2
BB 06/03 5
BB 07/03 8
BB 08/03 7
CC 05/03 3
CC 06/03 6
CC 07/03 9
CC 08/03 1
我该怎么做?
从宽到长的转换的原因是,在下一阶段,我想根据日期和初始列名(AA,BB,CC)将此数据框与另一个数据框合并。
解决方案 1:
使用pandas.melt
或pandas.DataFrame.melt
从宽转换为长:
df = pd.DataFrame({
'date' : ['05/03', '06/03', '07/03', '08/03'],
'AA' : [1, 4, 7, 5],
'BB' : [2, 5, 8, 7],
'CC' : [3, 6, 9, 1]
}).set_index('date')
df
AA BB CC
date
05/03 1 2 3
06/03 4 5 6
07/03 7 8 9
08/03 5 7 1
为了转换,我们只需要重置索引,然后融化:
df = df.reset_index()
pd.melt(df, id_vars='date', value_vars=['AA', 'BB', 'CC'])
使用.reset_index
after .melt
,无需指定value_vars
。
dfm = df.melt(ignore_index=False).reset_index()
最终结果 - 两种选择
date variable value
0 05/03 AA 1
1 06/03 AA 4
2 07/03 AA 7
3 08/03 AA 5
4 05/03 BB 2
5 06/03 BB 5
6 07/03 BB 8
7 08/03 BB 7
8 05/03 CC 3
9 06/03 CC 6
10 07/03 CC 9
11 08/03 CC 1
解决方案 2:
更新
正如 George Liu 在另一个答案中所展示的那样,pd.melt
这是解决此问题的惯用、灵活且快速的解决方案。请勿用于unstack
此。
unstack
返回具有多索引的系列:
In [38]: df.unstack()
Out[38]:
date
AA 05/03 1
06/03 4
07/03 7
08/03 5
BB 05/03 2
06/03 5
07/03 8
08/03 7
CC 05/03 3
06/03 6
07/03 9
08/03 1
dtype: int64
您可以在返回的系列上调用 reset_index:
In [39]: df.unstack().reset_index()
Out[39]:
level_0 date 0
0 AA 05-03 1
1 AA 06-03 4
2 AA 07-03 7
3 AA 08-03 5
4 BB 05-03 2
5 BB 06-03 5
6 BB 07-03 8
7 BB 08-03 7
8 CC 05-03 3
9 CC 06-03 6
10 CC 07-03 9
11 CC 08-03 1
或者构建一个具有多索引的数据框:
In [40]: pd.DataFrame(df.unstack())
Out[40]:
0
date
AA 05-03 1
06-03 4
07-03 7
08-03 5
BB 05-03 2
06-03 5
07-03 8
08-03 7
CC 05-03 3
06-03 6
07-03 9
08-03 1
解决方案 3:
除了unstack
和之外melt
,stack
这里也可以使用。
df1 = df.stack().reset_index(name='value')
# change "weird" column label
df1 = df.stack().reset_index(name='value').rename(columns={'level_1': 'variable'})
melt
、stack
和都是unstack
非常快的方法,因此在正常情况下,运行时差异几乎无关紧要。如果运行时是一个问题,也可以使用基于 numpy 的解决方案(比 快约 50% melt
)。这个想法是简单地将框架中的值展平为 1D 数组,并重复索引和列标签。
df1 = pd.DataFrame({ 'variable': np.tile(df.columns, len(df)), 'date': df.index.repeat(df.shape[1]), 'value': df.values.ravel()})
如果不需要将列标签作为单独的列,那么另一个非常快速的功能是pd.lreshape
。
df1 = pd.lreshape(df.reset_index(), {'value': ['AA', 'BB', 'CC']})
相关推荐
热门文章
项目管理软件有哪些?
热门标签
云禅道AD