Pandas實(shí)現(xiàn)Hive中的窗口函數(shù)

1论熙、Hive窗口函數(shù)

我們先來(lái)介紹一下Hive中幾個(gè)常見的窗口函數(shù)盯荤,row_number(),lag()和lead()揭蜒。

row_number()

該函數(shù)的格式如下:

row_Number() OVER (partition by 分組字段 ORDER BY 排序字段 排序方式asc/desc)

簡(jiǎn)單的說(shuō)彪见,我們使用partition by后面的字段對(duì)數(shù)據(jù)進(jìn)行分組丘薛,在每個(gè)組內(nèi),使用ORDER BY后面的字段進(jìn)行排序睦柴,并給每條記錄增加一個(gè)排序序號(hào)诽凌。

lag()

該函數(shù)的格式如下:

lag(字段名,N) over(partition by 分組字段 order by 排序字段 排序方式) 

lag括號(hào)里理由兩個(gè)參數(shù),第一個(gè)是字段名坦敌,第二個(gè)是數(shù)量N侣诵,這里的意思是,取分組排序之后比該條記錄序號(hào)小N的對(duì)應(yīng)記錄的指定字段的值狱窘,如果字段名為ts杜顺,N為1,就是取分組排序之后上一條記錄的ts值蘸炸。

lead()
該函數(shù)的格式如下:

lead(字段名,N) over(partition by 分組字段 order by 排序字段 排序方式) 

lead括號(hào)里理由兩個(gè)參數(shù)躬络,第一個(gè)是字段名,第二個(gè)是數(shù)量N搭儒,這里的意思是穷当,取分組排序之后比該條記錄序號(hào)大N的對(duì)應(yīng)記錄的對(duì)應(yīng)字段的值,如果字段名為ts淹禾,N為1馁菜,就是取分組排序之后下一條記錄的ts值。

有關(guān)這幾個(gè)函數(shù)的詳細(xì)的實(shí)例铃岔,可以參考我之前寫過(guò)的文章:http://www.reibang.com/p/3738d3591da9汪疮,這里我們就不再贅述。

2毁习、窗口函數(shù)的Pandas實(shí)現(xiàn)

接下來(lái)智嚷,我們介紹如何使用Pandas來(lái)實(shí)現(xiàn)上面的幾個(gè)窗口函數(shù)。

數(shù)據(jù)使用
我們建立如下的測(cè)試數(shù)據(jù)集:

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})

我們使用C作為分組列蜓洪,使用A作為窗口列纤勒。

2.1 row_number()

該函數(shù)的意思即分組排序,在pandas中我們可以結(jié)合groupby和rank函數(shù)來(lái)實(shí)現(xiàn)和row_number()類似的功能隆檀。

我們先看一下實(shí)現(xiàn)代碼:

df['row_number'] = df['A'].groupby(df['C']).rank(ascending=True,method='first')
print(df)

代碼的輸出為:

這樣我們的row_number功能就實(shí)現(xiàn)了,groupby方法大家應(yīng)該很熟悉了粹湃,那么我們主要介紹一下rank函數(shù)恐仑,rank函數(shù)主要有兩個(gè)參數(shù),首先是ascending參數(shù)为鳄,決定是按照升序還是降序排列裳仆,這里我們選擇的是升序。第二個(gè)參數(shù)是填充方式孤钦,主要有以下幾種方式:
dense:稠密的方式歧斟,即當(dāng)兩個(gè)或多個(gè)的數(shù)值相同時(shí)纯丸,使用同樣的序號(hào),同時(shí)后面的序號(hào)是該序號(hào)+1静袖,即多個(gè)相同的值只會(huì)占用一個(gè)序號(hào)位觉鼻,例如四個(gè)數(shù)的排序,中間兩個(gè)數(shù)相同队橙,那么四個(gè)數(shù)的排序?yàn)?坠陈,2,2捐康,3.

我們用代碼看一下效果:

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})
df['row_number'] = df['A'].groupby(df['C']).rank(ascending=True,method='min')
print(df)

輸出為:

first:即當(dāng)兩個(gè)或多個(gè)的數(shù)值相同時(shí)仇矾,使用不樣的序號(hào),按照數(shù)據(jù)出現(xiàn)的先后順序進(jìn)行排序解总,這個(gè)其實(shí)跟row_number的實(shí)現(xiàn)是相同的贮匕。

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})
df['row_number'] = df['A'].groupby(df['C']).rank(ascending=True,method='first')
print(df)

輸出為:

max :當(dāng)兩個(gè)或多個(gè)的數(shù)值相同時(shí),使用相同的序號(hào),不過(guò)使用的是能達(dá)到的最大的序號(hào)值花枫。例如四個(gè)數(shù)的排序粗合,中間兩個(gè)數(shù)相同,那么四個(gè)數(shù)的排序?yàn)?乌昔,3隙疚,3,4.

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})
df['row_number'] = df['A'].groupby(df['C']).rank(ascending=True,method='max')
print(df)

輸出為:

min :當(dāng)兩個(gè)或多個(gè)的數(shù)值相同時(shí),使用相同的序號(hào)磕道,不過(guò)使用的是能達(dá)到的最小的序號(hào)值供屉。例如四個(gè)數(shù)的排序,中間兩個(gè)數(shù)相同溺蕉,那么四個(gè)數(shù)的排序?yàn)?伶丐,2,2疯特,4.

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})
df['row_number'] = df['A'].groupby(df['C']).rank(ascending=True,method='min')
print(df)

輸出為:

2.2 lag/lead函數(shù)

pandas中使用shift函數(shù)來(lái)實(shí)現(xiàn)lag/lead函數(shù)哗魂,首先我們來(lái)看一個(gè)例子:

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})

df['lag'] = df.sort_values('A').groupby('C')['A'].shift(1)
df['lead'] = df.sort_values('A').groupby('C')['A'].shift(-1)
print(df)

輸出為:

可以看到,當(dāng)shift函數(shù)中的數(shù)字為正數(shù)時(shí)漓雅,我們就實(shí)現(xiàn)了lag的功能录别,當(dāng)數(shù)字為負(fù)數(shù)時(shí),實(shí)現(xiàn)的是lead的功能邻吞。不過(guò)這里切記组题,一定要排序哦,否則可能出現(xiàn)下面的結(jié)果:

df = pd.DataFrame({'A':[12,20,12,5,18,11,18],
                   'C':['A','B','A','B','B','A','A']})

df['lag'] = df.groupby('C')['A'].shift(1)
df['lead'] = df.groupby('C')['A'].shift(-1)
print(df)

輸出為抱冷,這個(gè)就是完全根據(jù)數(shù)據(jù)出現(xiàn)的順序進(jìn)行排序的崔列,不符合我們的要求!

好啦旺遮,兩個(gè)簡(jiǎn)單的小應(yīng)用就介紹到這里了赵讯,希望可以幫到大家盈咳。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市边翼,隨后出現(xiàn)的幾起案子鱼响,更是在濱河造成了極大的恐慌,老刑警劉巖讯私,帶你破解...
    沈念sama閱讀 221,576評(píng)論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件热押,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡斤寇,警方通過(guò)查閱死者的電腦和手機(jī)桶癣,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,515評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)娘锁,“玉大人牙寞,你說(shuō)我怎么就攤上這事∧眩” “怎么了间雀?”我有些...
    開封第一講書人閱讀 168,017評(píng)論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)镊屎。 經(jīng)常有香客問(wèn)我惹挟,道長(zhǎng),這世上最難降的妖魔是什么缝驳? 我笑而不...
    開封第一講書人閱讀 59,626評(píng)論 1 296
  • 正文 為了忘掉前任连锯,我火速辦了婚禮,結(jié)果婚禮上用狱,老公的妹妹穿的比我還像新娘运怖。我一直安慰自己,他們只是感情好夏伊,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,625評(píng)論 6 397
  • 文/花漫 我一把揭開白布摇展。 她就那樣靜靜地躺著,像睡著了一般溺忧。 火紅的嫁衣襯著肌膚如雪咏连。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,255評(píng)論 1 308
  • 那天砸狞,我揣著相機(jī)與錄音捻勉,去河邊找鬼。 笑死刀森,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的报账。 我是一名探鬼主播研底,決...
    沈念sama閱讀 40,825評(píng)論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼埠偿,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了榜晦?” 一聲冷哼從身側(cè)響起冠蒋,我...
    開封第一講書人閱讀 39,729評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎乾胶,沒(méi)想到半個(gè)月后抖剿,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,271評(píng)論 1 320
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡识窿,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,363評(píng)論 3 340
  • 正文 我和宋清朗相戀三年斩郎,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片喻频。...
    茶點(diǎn)故事閱讀 40,498評(píng)論 1 352
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡缩宜,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出甥温,到底是詐尸還是另有隱情锻煌,我是刑警寧澤,帶...
    沈念sama閱讀 36,183評(píng)論 5 350
  • 正文 年R本政府宣布姻蚓,位于F島的核電站宋梧,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏狰挡。R本人自食惡果不足惜捂龄,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,867評(píng)論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望圆兵。 院中可真熱鬧跺讯,春花似錦、人聲如沸殉农。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,338評(píng)論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)超凳。三九已至愈污,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間轮傍,已是汗流浹背暂雹。 一陣腳步聲響...
    開封第一講書人閱讀 33,458評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留创夜,地道東北人杭跪。 一個(gè)月前我還...
    沈念sama閱讀 48,906評(píng)論 3 376
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親涧尿。 傳聞我的和親對(duì)象是個(gè)殘疾皇子系奉,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,507評(píng)論 2 359