使用Pandas怎么實現重塑和軸向旋轉

這篇文章給大家介紹使用Pandas怎么實現重塑和軸向旋轉，內容非常詳細，感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助。

import numpy as np
import pandas as pd
from pandas import Series,DataFrame

一、重塑

• stack:將數據的列索引旋轉為行索引

• unstack:將數據的行索引旋轉為列索引

df = DataFrame({'水果':['蘋果','梨','草莓'],
        '數量':[3,4,5],
        '價格':[4,5,6]})
print(df)

   價格  數量  水果
0   4   3  蘋果
1   5   4   梨
2   6   5  草莓

1.stack()

stack_df = df.stack()
print(stack_df)

0  價格     4
   數量     3
   水果    蘋果
1  價格     5
   數量     4
   水果     梨
2  價格     6
   數量     5
   水果    草莓

dtype: object

2.unstack()

print(stack_df.unstack())

  價格 數量  水果
0  4  3  蘋果
1  5  4   梨
2  6  5  草莓

3.通過level參數指定旋轉軸的層次(默認level=-1)

print(stack_df.unstack(level=0))

     0  1   2
價格   4  5   6
數量   3  4   5
水果  蘋果  梨  草莓

二、軸向旋轉(pivot)

pivot(index,columns,values)：將index指定為行索引，columns是列索引，values則是DataFrame中的值

df = DataFrame({'水果種類':['蘋果','蘋果','梨','梨','草莓','草莓'],
        '信息':['價格','數量','價格','數量','價格','數量'],
        '值':[4,3,5,4,6,5]})
print(df)

   信息  值 水果種類
0  價格  4   蘋果
1  數量  3   蘋果
2  價格  5    梨
3  數量  4    梨
4  價格  6   草莓
5  數量  5   草莓

將水果種類作為行索引，將信息作為列索引

print(df.pivot('水果種類','信息','值'))

信息    價格  數量
水果種類       
梨      5   4
蘋果     4   3
草莓     6   5

pivot可以用set_index和unstack等價的實現

print(df.set_index(['水果種類','信息']).unstack())

      值  
信息   價格 數量
水果種類     
梨     5  4
蘋果    4  3
草莓    6  5

關于使用Pandas怎么實現重塑和軸向旋轉就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。