一文搞懂Pandas數據透視的4個函數的使用

大傢好,我是丁小傑!

今天和大傢分享Pandas中四種有關數據透視的通用函數,在數據處理中遇到這類需求時,能夠很好地應對。

pandas.melt()

melt函數的主要作用是將DataFrame從寬格式轉換成長格式。

pandas.melt(frame,id_vars=None, value_vars=None, var_name=None, value_name='value', col_level=None, ignore_index=True)

參數含義

  • id_vars:tuple, list, or ndarray,可選,作為標識符變量的列
  • value_vars:tuple, list, or ndarray, 可選,透視列,如果未指定,則使用未設置為id_vars的所有列。
  • var_name:scalar,默認為None,使用variable作為列名
  • value_name:標量, default ‘value’,value列的名稱
  • col_level:int or str, 可選,如果列是多層索引,melt將應用於指定級別
  • ignore_index:bool, 默認為True,相當於從0開始重新排序。如果為False,則保留原來的索引,索引標簽將出現重復。

看個例子先:

import pandas as pd

df = pd.DataFrame(
    {'地區': ['A', 'B', 'C'],
     '2020': [80, 60, 40],
     '2021': [800, 600, 400], 
     '2022': [8000, 6000, 4000]})

pd.melt(df,
        id_vars=['地區'],
        value_vars=['2020', '2021', '2022'])

設置var_namevalue_name

df = pd.melt(df,
             id_vars=['地區'],
             value_vars=['2020', '2021', '2022'],
             var_name='年份',
             value_name='銷售額')

pandas.pivot()

pivot函數主要用於通過索引及列值對DataFrame重構。

pandas.pivot(data, index=None, columns=None, values=None)

參數含義

  • data:DataFrame對象
  • index:可選,用於新DataFrame的索引
  • columns:用於創建新DataFrame的列
  • values:可選,用於填充新DataFrame的值

用上面的結果舉個例子:

df.pivot(index='年份',
         columns='地區',
         values='銷售額')

也可以寫成以下格式。

df.pivot(index='年份', columns='地區')['銷售額']

添加一個銷量列,同時統計兩個values,這樣會使columns變成多層索引。

df['銷量'] = df['銷售額']/10
df.pivot(index='年份',
         columns='地區',
         values=['銷售額', '銷量'])

添加一個月份列,指定兩個index

df['月份'] = [f'{m}月' for m in range(1, 4)]*3
df.pivot(index=['年份', '月份'],
         columns='地區',
         values='銷售額')

使用pivot時需要註意,當indexcolumns出現重復時,會導致ValueError

df = pd.DataFrame(
        {'地區': ['A', 'A', 'B', 'C'],
         '年份': ['2020', '2020', '2021', '2022'],
         '銷售額': [800, 600, 400, 200]})

df.pivot(index='地區',
         columns='年份',
         values='銷售額')
# ValueError

pandas.pivot_table()

這個函數之前已經單獨講過瞭,詳見Pandas玩轉數據透視表,相比於pivotpivot_table的靈活性更強。

pandas.crosstab()

crosstab函數計算兩個(或多個)數組的簡單交叉表。默認情況下計算元素的頻率表。

pandas.crosstab(index, columns, values=None, rownames=None, colnames=None, aggfunc=None, margins=False, margins_name='All', dropna=True, normalize=False)

看下例子:

這裡默認計算頻率。

import numpy as np
array_A = np.array(["one", "two", "two", "three", "three", "three"], dtype=object)
array_B = np.array(["Python", "Python", "Python", "C", "C", "C"], dtype=object)
array_C = np.array(["Y", "Y", "Y", "N", "N", "N"])
pd.crosstab(array_A,
           [array_B, array_C],
           rownames=['array_A'],
           colnames=['array_B', 'array_C'])

新建一個values列,計算總和。

array_D = np.array([1, 4, 9, 16, 25, 36])
pd.crosstab(index=array_A,
            columns=[array_B, array_C],
            rownames=['array_A'],
            colnames=['array_B', 'array_C'],
            values=array_D,
            aggfunc='sum')

到此這篇關於一文搞懂Pandas數據透視的4個函數的使用的文章就介紹到這瞭,更多相關Pandas數據透視內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: