Python unstack函數(shù)有替代方法嗎

小樊
84
2024-07-20 18:08:02

在Python中，可以使用pivot_table函數(shù)來(lái)替代unstack函數(shù)的功能。pivot_table函數(shù)可以根據(jù)指定的行和列索引對(duì)數(shù)據(jù)進(jìn)行重塑，類似于unstack函數(shù)的功能。示例代碼如下：

import pandas as pd

df = pd.DataFrame({
    'A': ['foo', 'foo', 'foo', 'bar', 'bar', 'bar'],
    'B': ['one', 'one', 'two', 'two', 'one', 'one'],
    'C': [1, 2, 3, 4, 5, 6],
    'D': [10, 20, 30, 40, 50, 60]
})

# 使用pivot_table函數(shù)對(duì)數(shù)據(jù)進(jìn)行重塑
df_pivot = df.pivot_table(index='A', columns='B', values='C', aggfunc='sum')
print(df_pivot)

上述代碼中，先創(chuàng)建了一個(gè)包含’A’、‘B’、‘C’、'D’列的DataFrame，然后使用pivot_table函數(shù)將數(shù)據(jù)按照’A’列和’B’列進(jìn)行重塑，計(jì)算’C’列的和作為新的值。通過(guò)這種方式可以達(dá)到和unstack函數(shù)類似的效果。

0 踩