Pandas 必知必會的18個實用技巧珊搀,值得收藏!

干凈整潔的數(shù)據(jù)是后續(xù)進(jìn)行研究和分析的基礎(chǔ)尾菇。數(shù)據(jù)科學(xué)家們會花費大量的時間來清理數(shù)據(jù)集境析,毫不夸張地說,數(shù)據(jù)清洗會占據(jù)他們80%的工作時間桌吃,而真正用來分析數(shù)據(jù)的時間只占到20%左右俐银。

所以兰伤,數(shù)據(jù)清洗到底是在清洗些什么?
通常來說沛鸵,你所獲取到的原始數(shù)據(jù)不能直接用來分析,因為它們會有各種各樣的問題缆八,如包含無效信息谒臼,列名不規(guī)范、格式不一致耀里,存在重復(fù)值蜈缤,缺失值,異常值等.....

本文會給大家介紹一些Python中自帶的Pandas和NumPy庫進(jìn)行數(shù)據(jù)清洗的實用技巧冯挎。

一底哥、 read_csv 讀取文件

這是讀取數(shù)據(jù)的入門級命令,在分析一個數(shù)據(jù)集的時候房官,很多信息其實是用不到的趾徽,因此,需要去除不必要的行或列翰守。這里以csv文件為例孵奶,在導(dǎo)入的時候就可以通過設(shè)置pd.read_csv()里面的參數(shù)來實現(xiàn)這個目的。



先來感受一下官方文檔中給出的詳細(xì)解釋蜡峰,里面的參數(shù)是相當(dāng)?shù)亩嗔嗽疚闹唤榻B比較常用的幾個朗恳,感興趣的話,可以好好研究一下文檔载绿,這些參數(shù)還是非常好用的粥诫,能省去很多導(dǎo)入后整理的工作。

【header】默認(rèn)header=0崭庸,即將文件中的0行作為列名和數(shù)據(jù)的開頭怀浆,但有時候0行的數(shù)據(jù)是無關(guān)的,我們想跳過0行怕享,讓1行作為數(shù)據(jù)的開頭执赡,可以通過將header設(shè)置為1來實現(xiàn)。

【usecols】根據(jù)列的位置或名字函筋,如[0,1,2]或[‘a(chǎn)’, ‘b’, ‘c’]搀玖,選出特定的列。

【nrows】要導(dǎo)入的數(shù)據(jù)行數(shù)驻呐,在數(shù)據(jù)量很大灌诅、但只想導(dǎo)入其中一部分時使用。

【names】:指定自定義列名含末。 此列表中不允許有重復(fù)項猜拾。

【index_col】: 指定用作數(shù)據(jù)框的行標(biāo)簽的列,以字符串名稱或列索引的形式給出

【na_values】:指定數(shù)據(jù)框中需要被識別為NA/NaN 的字符串或字符串列表
當(dāng)一列缺失時直接指定 na_values='-1'
當(dāng)多列缺失時佣盒,增加字典的鍵值對即可 na_values={'a':'-1','b':'-10'}

二挎袜、重新命名列

當(dāng)原始數(shù)據(jù)的列名不好理解,或者不夠簡潔時肥惭,可以用.rename()方法進(jìn)行修改盯仪。這里我們把英文的列名改成中文,先創(chuàng)建一個字典蜜葱,把要修改的列名定義好全景,然后調(diào)用rename()方法。

new_names = {'舊列名': '新列名'}
df.rename(columns=new_names, inplace=True)

三牵囤、重新設(shè)置索引

數(shù)據(jù)默認(rèn)的索引是從0開始的有序整數(shù)爸黄,但如果想把某一列設(shè)置為新的索引,除了可以用read_csv()里的參數(shù)index_col揭鳞,還可以用.set_index()方法實現(xiàn)炕贵。

df.set_index('列名', inplace=True)

另外補充,如果數(shù)據(jù)經(jīng)過刪除或結(jié)構(gòu)調(diào)整后野崇,我們可以重置索引称开,讓索引從0開始,依次排序乓梨。

df3.reset_index(drop=True)

四鳖轰、用字符串操作規(guī)范列

字符串str操作是非常實用的清酥,因為列中總是會包含不必要的字符,常用的方法如下:

lower()
upper()

str.lower() 是把大寫轉(zhuǎn)換成小寫脆霎,同理,str.upper()是把小寫轉(zhuǎn)換成大寫狈惫,將示例中用大寫字母表示的索引轉(zhuǎn)換成小寫睛蛛。

capitalize()

設(shè)置首字母大寫

replace()

str.replace("a", "") 替換特定字符。這里把列中的a去掉胧谈,替換成空字符忆肾。

strip()

去除字符串中的頭尾空格、以及\n \t菱肖。

split()

str.split('x') 使用字符串中的'x'字符作為分隔符客冈,將字符串分隔成列表。這里將列中的值以'.'進(jìn)行分割稳强。

get()

str.get() 選取列表中某個位置的值场仲。接著上面分割后的結(jié)果,我們用str.get(0)取出列表中前一個位置的數(shù)值退疫,生成新的一列渠缕。

contains()

str.contains() 判斷是否存在某個字符,返回的是布爾值褒繁。

find()

str.find("-")檢測字符串中是否包含"-"亦鳞,如果包含,則返回該子字符串開始位置的索引值;如果不包含棒坏,則返回-1燕差。

學(xué)完基本的字符串操作方法,我們來看一下如何結(jié)合NumPy來提高字符串操作的效率坝冕。
我們可以將Pandas中的.str()方法與NumPy的np.where函數(shù)相結(jié)合徒探,np.where函數(shù)是Excel的IF()宏的矢量化形式,它的語法如下:

np.where(condition, then, else)

如果condition條件為真喂窟,則執(zhí)行then刹帕,否則執(zhí)行else。這里的condition條件可以是一個類數(shù)組的對象谎替,也可以是一個布爾表達(dá)式偷溺,我們也可以利用np.where函數(shù)嵌套多個條件進(jìn)行矢量化計算和判斷。

np.where(condition1, x1, 
        np.where(condition2, x2, 
            np.where(condition3, x3, ...)))

五钱贯、自定義函數(shù)規(guī)范列

接下來就要對列中的字符串進(jìn)行整理挫掏,除了利用循環(huán)和.str()方法相結(jié)合的方式進(jìn)行操作,我們還可以選擇用applymap()方法秩命,它會將傳入的函數(shù)作用于整個DataFrame所有行列中的每個元素尉共。



先定義函數(shù)get_citystate(item)褒傅,功能是只提取元素中的有效信息。然后袄友,我們將這個函數(shù)傳入applymap()殿托,并應(yīng)用于df3,看起來是不是干凈多了剧蚣,結(jié)果如下:


六支竹、copy

如果你沒聽說過它的話,我不得強調(diào)它的重要性鸠按。輸入下面的命令:

import pandas as pd
df1 = pd.DataFrame({  a :[0,0,0],  b : [1,1,1]})
df2 = df1
df2[ a ] = df2[ a ] + 1
df1.head()

你會發(fā)現(xiàn)df1已經(jīng)發(fā)生了改變礼搁。這是因為df2 = df1并不是生成一個df1的復(fù)制品并把它賦值給df2,而是設(shè)定一個指向df1的指針目尖。所以只要是針對df2的改變馒吴,也會相應(yīng)地作用在df1上。為了解決這個問題瑟曲,你既可以這樣做:

df2 = df1.copy()

也可以這樣做:

from copy import deepcopy
df2 = deepcopy(df1)

七饮戳、value_counts()

這個命令用于檢查值的分布。你想要檢查下“c”列中出現(xiàn)的值以及每個值所出現(xiàn)的頻率洞拨,可以使用:

df[ c ].value_counts()

下面是一些有用的小技巧/參數(shù):

normalize = True:查看每個值出現(xiàn)的頻率而不是頻次數(shù)莹捡。

dropna = False: 把缺失值也保留在這次統(tǒng)計中。

sort = False: 將數(shù)據(jù)按照值來排序而不是按照出現(xiàn)次數(shù)排序扣甲。

df[‘c].value_counts().reset_index(): 將這個統(tǒng)計表轉(zhuǎn)換成pandas的dataframe并且進(jìn)行處理篮赢。

八、lsin () 琉挖,依據(jù)指定ID來選取行

lsin () 用于過濾數(shù)據(jù)幀启泣。Isin () 有助于選擇特定列中具有特定(或多個)值的行。
在SQL中我們可以使用 SELECT * FROM … WHERE ID in (‘A001’,‘C022’, …)來獲取含有指定ID的記錄示辈。如果你也想在Pandas中做類似的事情寥茫,你可以使用:

df_filter = df[ ID ].isin([ A001 , C022 ,...])
df[df_filter]

九、select_dtypes()

select_dtypes() 的作用是矾麻,基于 dtypes 的列返回數(shù)據(jù)幀列的一個子集纱耻。這個函數(shù)的參數(shù)可設(shè)置為包含所有擁有特定數(shù)據(jù)類型的列,亦或者設(shè)置為排除具有特定數(shù)據(jù)類型的列险耀。

# We ll use the same dataframe that we used for read_csv
framex =  df.select_dtypes(include="float64")# Returns only time column

十弄喘、pivot_table( )

pivot_table( ) 也是 Pandas 中一個非常有用的函數(shù)。如果對 pivot_table( ) 在 excel 中的使用有所了解甩牺,那么就非常容易上手了蘑志。

pivot_table(data, values=None, index=None, columns=None,aggfunc='mean', fill_value=None, margins=False, dropna=True, margins_name='All')

pivot_table有四個最重要的參數(shù)index、values、columns急但、aggfunc

  • Index就是層次字段澎媒,要通過透視表獲取什么信息就按照相應(yīng)的順序設(shè)置字段
  • Values可以對需要的計算數(shù)據(jù)進(jìn)行篩選
  • Columns類似Index可以設(shè)置列層次字段,它不是一個必要參數(shù)波桩,作為一種分割數(shù)據(jù)的可選方式
  • aggfunc參數(shù)可以設(shè)置我們對數(shù)據(jù)聚合時進(jìn)行的函數(shù)操作
# Create a sample dataframe
school = pd.DataFrame({ "A" : [ "Jay" ,  "Usher" ,  "Nicky" , " Romero" ,  "Will" ], 
       "B" : [ "Masters" ,  "Graduate",  "Graduate" ,  ''Masters" ,  "Graduate" ], 
       "C" : [26, 22, 20, 23, 24]})# Lets create a pivot table to segregate students based on age and course
table = pd.pivot_table(school, values = "A" , index =[ "B" ,  "C" ], 
                         columns =[ "B" ], aggfunc = np.sum, fill_value="Not Available") 

table

十一戒努、計算變量缺失率

df=pd.read_csv('train.csv')
def missing_cal(df):
    """
    df :數(shù)據(jù)集
    
    return:每個變量的缺失率
    """
    missing_series = df.isnull().sum()/df.shape[0]
    missing_df = pd.DataFrame(missing_series).reset_index()
    missing_df = missing_df.rename(columns={'index':'col',
                                            0:'missing_pct'})
    missing_df = missing_df.sort_values('missing_pct',ascending=False).reset_index(drop=True)
    return missing_df
missing_cal(df)

如果需要計算樣本的缺失率分布,只要加上參數(shù)axis=1

十二镐躲、獲取分組里最大值所在的行方法

分為分組中有重復(fù)值和無重復(fù)值兩種储玫。無重復(fù)值的情況。

df = pd.DataFrame({'Sp':['a','b','c','d','e','f'], 'Mt':['s1', 's1', 's2','s2','s2','s3'], 'Value':[1,2,3,4,5,6], 'Count':[3,2,5,10,10,6]})
df

df.iloc[df.groupby(['Mt']).apply(lambda x: x['Count'].idxmax())]

先按Mt列進(jìn)行分組匀油,然后對分組之后的數(shù)據(jù)框使用idxmax函數(shù)取出Count最大值所在的列缘缚,再用iloc位置索引將行取出勾笆。有重復(fù)值的情況

df["rank"] = df.groupby("ID")["score"].rank(method="min", ascending=False).astype(np.int64)
df[df["rank"] == 1][["ID", "class"]]

對ID進(jìn)行分組之后再對分?jǐn)?shù)應(yīng)用rank函數(shù)敌蚜,分?jǐn)?shù)相同的情況會賦予相同的排名,然后取出排名為1的數(shù)據(jù)窝爪。

十三弛车、多列合并為一行

df = pd.DataFrame({'id_part':['a','b','c','d'], 'pred':[0.1,0.2,0.3,0.4], 'pred_class':['women','man','cat','dog'], 'v_id':['d1','d2','d3','d1']})

df.groupby(['v_id']).agg({'pred_class': [', '.join],'pred': lambda x: list(x),
'id_part': 'first'}).reset_index()

十四、組內(nèi)排序

df = pd.DataFrame([['A',1],['A',3],['A',2],['B',5],['B',9]], columns = ['name','score'])

介紹兩種高效地組內(nèi)排序的方法蒲每。

df.sort_values(['name','score'], ascending = [True,False])
df.groupby('name').apply(lambda x: x.sort_values('score', ascending=False)).reset_index(drop=True)

十五纷跛、選擇特定類型的列

drinks = pd.read_csv('data/drinks.csv')
# 選擇所有數(shù)值型的列
drinks.select_dtypes(include=['number']).head()
# 選擇所有字符型的列
drinks.select_dtypes(include=['object']).head()
drinks.select_dtypes(include=['number','object','category','datetime']).head()
# 用 exclude 關(guān)鍵字排除指定的數(shù)據(jù)類型
drinks.select_dtypes(exclude=['number']).head()

十六、字符串轉(zhuǎn)換為數(shù)值

df = pd.DataFrame({'列1':['1.1','2.2','3.3'],
                  '列2':['4.4','5.5','6.6'],
                  '列3':['7.7','8.8','-']})
df
df.astype({'列1':'float','列2':'float'}).dtypes

用這種方式轉(zhuǎn)換第三列會出錯邀杏,因為這列里包含一個代表 0 的下劃線贫奠,pandas 無法自動判斷這個下劃線。為了解決這個問題望蜡,可以使用 to_numeric() 函數(shù)來處理第三列唤崭,讓 pandas 把任意無效輸入轉(zhuǎn)為 NaN。
df = df.apply(pd.to_numeric, errors='coerce').fillna(0)

十七脖律、優(yōu)化 DataFrame 對內(nèi)存的占用

方法一:只讀取切實所需的列谢肾,使用usecols參數(shù)

cols = ['beer_servings','continent']
small_drinks = pd.read_csv('data/drinks.csv', usecols=cols)

方法二:把包含類別型數(shù)據(jù)的 object 列轉(zhuǎn)換為 Category 數(shù)據(jù)類型,通過指定 dtype 參數(shù)實現(xiàn)小泉。

dtypes ={'continent':'category'}
smaller_drinks = pd.read_csv('data/drinks.csv',usecols=cols, dtype=dtypes)

十八芦疏、把字符串分割為多列

df = pd.DataFrame({'姓名':['張 三','李 四','王 五'],
                   '所在地':['北京-東城區(qū)','上海-黃浦區(qū)','廣州-白云區(qū)']})
df
df.姓名.str.split(' ', expand=True)
11.把 Series 里的列表轉(zhuǎn)換為 DataFrame
df = pd.DataFrame({'列1':['a','b','c'],'列2':[[10,20], [20,30], [30,40]]})
df

df_new = df.列2.apply(pd.Series)
pd.concat([df,df_new], axis='columns')

希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作能帶來一定的幫助,每天進(jìn)步一點點微姊,加油~

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末酸茴,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子兢交,更是在濱河造成了極大的恐慌弊决,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,324評論 6 498
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異飘诗,居然都是意外死亡与倡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,356評論 3 392
  • 文/潘曉璐 我一進(jìn)店門昆稿,熙熙樓的掌柜王于貴愁眉苦臉地迎上來纺座,“玉大人,你說我怎么就攤上這事溉潭【幌欤” “怎么了?”我有些...
    開封第一講書人閱讀 162,328評論 0 353
  • 文/不壞的土叔 我叫張陵喳瓣,是天一觀的道長馋贤。 經(jīng)常有香客問我,道長畏陕,這世上最難降的妖魔是什么配乓? 我笑而不...
    開封第一講書人閱讀 58,147評論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮惠毁,結(jié)果婚禮上犹芹,老公的妹妹穿的比我還像新娘。我一直安慰自己鞠绰,他們只是感情好腰埂,可當(dāng)我...
    茶點故事閱讀 67,160評論 6 388
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著蜈膨,像睡著了一般屿笼。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上翁巍,一...
    開封第一講書人閱讀 51,115評論 1 296
  • 那天驴一,我揣著相機與錄音,去河邊找鬼曙咽。 笑死蛔趴,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的例朱。 我是一名探鬼主播孝情,決...
    沈念sama閱讀 40,025評論 3 417
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼洒嗤!你這毒婦竟也來了箫荡?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,867評論 0 274
  • 序言:老撾萬榮一對情侶失蹤渔隶,失蹤者是張志新(化名)和其女友劉穎羔挡,沒想到半個月后洁奈,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,307評論 1 310
  • 正文 獨居荒郊野嶺守林人離奇死亡绞灼,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,528評論 2 332
  • 正文 我和宋清朗相戀三年利术,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片低矮。...
    茶點故事閱讀 39,688評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡印叁,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出军掂,到底是詐尸還是另有隱情轮蜕,我是刑警寧澤,帶...
    沈念sama閱讀 35,409評論 5 343
  • 正文 年R本政府宣布蝗锥,位于F島的核電站跃洛,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏终议。R本人自食惡果不足惜汇竭,卻給世界環(huán)境...
    茶點故事閱讀 41,001評論 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望痊剖。 院中可真熱鬧韩玩,春花似錦垒玲、人聲如沸陆馁。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,657評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽叮贩。三九已至,卻和暖如春佛析,著一層夾襖步出監(jiān)牢的瞬間益老,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,811評論 1 268
  • 我被黑心中介騙來泰國打工寸莫, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留捺萌,地道東北人。 一個月前我還...
    沈念sama閱讀 47,685評論 2 368
  • 正文 我出身青樓膘茎,卻偏偏與公主長得像桃纯,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子披坏,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,573評論 2 353