使用Python Pandas處理億級(jí)數(shù)據(jù)的方法
在數(shù)據(jù)分析領(lǐng)域,最熱門的莫過于Python和R語言,此前有一篇文章《別老扯什么Hadoop了,你的數(shù)據(jù)根本不夠大》指出:只有在超過5TB數(shù)據(jù)量的規(guī)模下,Hadoop才是一個(gè)合理的技術(shù)選擇。這次拿到近億條日志數(shù)據(jù),千萬級(jí)數(shù)據(jù)已經(jīng)是關(guān)系型數(shù)據(jù)庫的查詢分析瓶頸,之前使用過Hadoop對(duì)大量文本進(jìn)行分類,這次決定采用Python來處理數(shù)據(jù):
硬件環(huán)境
- CPU:3.5 GHz Intel Core i7
- 內(nèi)存:32 GB HDDR 3 1600 MHz
- 硬盤:3 TB Fusion Drive
數(shù)據(jù)分析工具
- Python:2.7.6
- Pandas:0.15.0
- IPython notebook:2.0.0
源數(shù)據(jù)如下表所示:
| Table | Size | Desc | |
|---|---|---|---|
| ServiceLogs | 98,706,832 rows x 14 columns | 8.77 GB | 交易日志數(shù)據(jù),每個(gè)交易會(huì)話可以有多條交易 |
| ServiceCodes | 286 rows × 8 columns | 20 KB | 交易分類的字典表 |
數(shù)據(jù)讀取
啟動(dòng)IPython notebook,加載pylab環(huán)境:
ipython notebook --pylab=inline
Pandas提供了IO工具可以將大文件分塊讀取,測試了一下性能,完整加載9800萬條數(shù)據(jù)也只需要263秒左右,還是相當(dāng)不錯(cuò)了。
import pandas as pd
reader = pd.read_csv('data/servicelogs', iterator=True)
try:
df = reader.get_chunk(100000000)
except StopIteration:
print "Iteration is stopped."
| 1百萬條 | 1千萬條 | 1億條 | |
|---|---|---|---|
| ServiceLogs | 1 s | 17 s | 263 s |
使用不同分塊大小來讀取再調(diào)用pandas.concat連接DataFrame,chunkSize設(shè)置在1000萬條左右速度優(yōu)化比較明顯。
loop = True
chunkSize = 100000
chunks = []
while loop:
try:
chunk = reader.get_chunk(chunkSize)
chunks.append(chunk)
except StopIteration:
loop = False
print "Iteration is stopped."
df = pd.concat(chunks, ignore_index=True)
下面是統(tǒng)計(jì)數(shù)據(jù),Read Time是數(shù)據(jù)讀取時(shí)間,Total Time是讀取和Pandas進(jìn)行concat操作的時(shí)間,根據(jù)數(shù)據(jù)總量來看,對(duì)5~50個(gè)DataFrame對(duì)象進(jìn)行合并,性能表現(xiàn)比較好。

如果使用Spark提供的Python Shell,同樣編寫Pandas加載數(shù)據(jù),時(shí)間會(huì)短25秒左右,看來Spark對(duì)Python的內(nèi)存使用都有優(yōu)化。
數(shù)據(jù)清洗
Pandas提供了DataFrame.describe方法查看數(shù)據(jù)摘要,包括數(shù)據(jù)查看(默認(rèn)共輸出首尾60行數(shù)據(jù))和行列統(tǒng)計(jì)。由于源數(shù)據(jù)通常包含一些空值甚至空列,會(huì)影響數(shù)據(jù)分析的時(shí)間和效率,在預(yù)覽了數(shù)據(jù)摘要后,需要對(duì)這些無效數(shù)據(jù)進(jìn)行處理。
首先調(diào)用DataFrame.isnull()方法查看數(shù)據(jù)表中哪些為空值,與它相反的方法是DataFrame.notnull(),Pandas會(huì)將表中所有數(shù)據(jù)進(jìn)行null計(jì)算,以True/False作為結(jié)果進(jìn)行填充,如下圖所示:

Pandas的非空計(jì)算速度很快,9800萬數(shù)據(jù)也只需要28.7秒。得到初步信息之后,可以對(duì)表中空列進(jìn)行移除操作。嘗試了按列名依次計(jì)算獲取非空列,和DataFrame.dropna()兩種方式,時(shí)間分別為367.0秒和345.3秒,但檢查時(shí)發(fā)現(xiàn) dropna() 之后所有的行都沒有了,查了Pandas手冊,原來不加參數(shù)的情況下, dropna() 會(huì)移除所有包含空值的行。如果只想移除全部為空值的列,需要加上 axis 和 how 兩個(gè)參數(shù):
df.dropna(axis=1, how='all')
共移除了14列中的6列,時(shí)間也只消耗了85.9秒。
接下來是處理剩余行中的空值,經(jīng)過測試,在DataFrame.replace()中使用空字符串,要比默認(rèn)的空值NaN節(jié)省一些空間;但對(duì)整個(gè)CSV文件來說,空列只是多存了一個(gè)“,”,所以移除的9800萬 x 6列也只省下了200M的空間。進(jìn)一步的數(shù)據(jù)清洗還是在移除無用數(shù)據(jù)和合并上。
對(duì)數(shù)據(jù)列的丟棄,除無效值和需求規(guī)定之外,一些表自身的冗余列也需要在這個(gè)環(huán)節(jié)清理,比如說表中的流水號(hào)是某兩個(gè)字段拼接、類型描述等,通過對(duì)這些數(shù)據(jù)的丟棄,新的數(shù)據(jù)文件大小為4.73GB,足足減少了4.04G!
數(shù)據(jù)處理
使用DataFrame.dtypes可以查看每列的數(shù)據(jù)類型,Pandas默認(rèn)可以讀出int和float64,其它的都處理為object,需要轉(zhuǎn)換格式的一般為日期時(shí)間。DataFrame.astype()方法可對(duì)整個(gè)DataFrame或某一列進(jìn)行數(shù)據(jù)格式轉(zhuǎn)換,支持Python和NumPy的數(shù)據(jù)類型。
df['Name'] = df['Name'].astype(np.datetime64)
對(duì)數(shù)據(jù)聚合,我測試了 DataFrame.groupby 和 DataFrame.pivot_table 以及 pandas.merge ,groupby 9800萬行 x 3列的時(shí)間為99秒,連接表為26秒,生成透視表的速度更快,僅需5秒。
df.groupby(['NO','TIME','SVID']).count() # 分組
fullData = pd.merge(df, trancodeData)[['NO','SVID','TIME','CLASS','TYPE']] # 連接
actions = fullData.pivot_table('SVID', columns='TYPE', aggfunc='count') # 透視表
根據(jù)透視表生成的交易/查詢比例餅圖:

將日志時(shí)間加入透視表并輸出每天的交易/查詢比例圖:
total_actions = fullData.pivot_table('SVID', index='TIME', columns='TYPE', aggfunc='count')
total_actions.plot(subplots=False, figsize=(18,6), kind='area')

除此之外,Pandas提供的DataFrame查詢統(tǒng)計(jì)功能速度表現(xiàn)也非常優(yōu)秀,7秒以內(nèi)就可以查詢生成所有類型為交易的數(shù)據(jù)子表:
tranData = fullData[fullData['Type'] == 'Transaction']
該子表的大小為[10250666 rows x 5 columns]。在此已經(jīng)完成了數(shù)據(jù)處理的一些基本場景。實(shí)驗(yàn)結(jié)果足以說明,在非“>5TB”數(shù)據(jù)的情況下,Python的表現(xiàn)已經(jīng)能讓擅長使用統(tǒng)計(jì)分析語言的數(shù)據(jù)分析師游刃有余。
以上就是本文的全部內(nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
在python中實(shí)現(xiàn)調(diào)用可執(zhí)行文件.exe的3種方法
今天小編就為大家分享一篇在python中實(shí)現(xiàn)調(diào)用可執(zhí)行文件.exe的3種方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07
詳解pandas.DataFrame中刪除包涵特定字符串所在的行
這篇文章主要介紹了pandas.DataFrame中刪除包涵特定字符串所在的行,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-04-04
Windows10下 python3.7 安裝 facenet的教程
這篇文章主要介紹了Windows10 python3.7 安裝 facenet的教程,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-09-09
在Python中append以及extend返回None的例子
今天小編就為大家分享一篇在Python中append以及extend返回None的例子,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-07-07
Python?pandas?DataFrame數(shù)據(jù)拼接方法
我們都知道在使用pandas處理數(shù)據(jù)的時(shí)候,往往會(huì)需要合并兩個(gè)或者多個(gè)DataFrame的操作,下面這篇文章主要給大家介紹了關(guān)于Python?pandas?DataFrame數(shù)據(jù)拼接方法的相關(guān)資料,需要的朋友可以參考下2022-07-07
Python實(shí)現(xiàn)自動(dòng)玩貪吃蛇程序
這篇文章主要介紹了通過Python實(shí)現(xiàn)的簡易的自動(dòng)玩貪吃蛇游戲的小程序,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)一學(xué)2022-01-01

