本文涉及pandas最常用的36個函數(shù),通過這些函數(shù)介紹如何完成數(shù)據(jù)生成和導(dǎo)入兽叮、數(shù)據(jù)清洗、預(yù)處理猾愿,以及最常見的數(shù)據(jù)分類鹦聪,數(shù)據(jù)篩選,分類匯總蒂秘,透視等最常見的操作泽本。
生成數(shù)據(jù)表
常見的生成數(shù)據(jù)表的方法有兩種,第一種是導(dǎo)入外部數(shù)據(jù)姻僧,第二
種是直接寫入數(shù)據(jù)规丽。Excel中的“文件”菜單中提供了獲取外部數(shù)據(jù)的功能,支持?jǐn)?shù)據(jù)庫和文本文件和頁面的多種數(shù)據(jù)源導(dǎo)入撇贺。
Python支持從多種類型的數(shù)據(jù)導(dǎo)入赌莺。在開始使用Python進行數(shù)據(jù)
導(dǎo)入前需要先導(dǎo)入pandas庫,為了方便起見松嘶,我們也同時導(dǎo)入numpy
庫.
import numpy as np
import pandas as pd
導(dǎo)入外部數(shù)據(jù)
df=pd.DataFrame(pd.read_csv('name.csv',header=1))
df=pd.DataFrame(pd.read_Excel('name.xlsx'))c
里面有很多可選參數(shù)設(shè)置艘狭,例如列名稱、索引列、數(shù)據(jù)格式等
直接寫入數(shù)據(jù)
df = pd.DataFrame({"id":[1001,1002,1003,1004,1005,1006],
"date":pd.date_range('20130102', periods=6),
"city":['Beijing ', 'SH', ' guangzhou ', 'Shen
zhen', 'shanghai', 'BEIJING '],
"age":[23,44,54,32,34,32],
"category":['100-A','100-B','110-A','110-C','2
10-A','130-F'],
"price":[1200,np.nan,2133,5433,np.nan,4432]},
columns =['id','date','city','category','age',
'price'])
數(shù)據(jù)表檢查
數(shù)據(jù)表檢查的目的是了解數(shù)據(jù)表的整體情況巢音,獲得數(shù)據(jù)表的關(guān)鍵信息遵倦、數(shù)據(jù)的概況,例如整個數(shù)據(jù)表的大小官撼、所占空間梧躺、數(shù)據(jù)格式、是否有
空值和重復(fù)項和具體的數(shù)據(jù)內(nèi)容傲绣,為后面的清洗和預(yù)處理做好準(zhǔn)備燥狰。
1.數(shù)據(jù)維度(行列)
Excel中可以通過CTRL+向下的光標(biāo)鍵,和CTRL+向右的光標(biāo)鍵
來查看行號和列號斜筐。Python中使用shape函數(shù)來查看數(shù)據(jù)表的維度龙致,也就是行數(shù)和列數(shù)。
df.shape
2.數(shù)據(jù)表信息
使用info函數(shù)查看數(shù)據(jù)表的整體信息顷链,包括數(shù)據(jù)維度目代、列名稱、數(shù)據(jù)格式和所占空間等信息嗤练。
數(shù)據(jù)表信息
df.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 6 entries, 0 to 5
Data columns (total 6 columns):
id 6 non-null int64
date 6 non-null datetime64[ns]
city 6 non-null object
category 6 non-null object
age 6 non-null int64
price 4 non-null float64
dtypes: datetime64[ns](1), float64(1), int64(2), object(2)
memory usage: 368.0+ bytes
3.查看數(shù)據(jù)格式
Excel中通過選中單元格并查看開始菜單中的數(shù)值類型來判斷數(shù)
據(jù)的格式榛了。Python中使用dtypes函數(shù)來返回數(shù)據(jù)格式。
Dtypes是一個查看數(shù)據(jù)格式的函數(shù)煞抬,可以一次性查看數(shù)據(jù)表中所
有數(shù)據(jù)的格式霜大,也可以指定一列來單獨查看
#查看數(shù)據(jù)表各列格式
df.dtypes
id int64
date datetime64[ns]
city object
category object
age int64
price float64
dtype: object
#查看單列格式
df['B'].dtype
dtype('int64')
4.查看空值
Excel中查看空值的方法是使用“定位條件”在“開始”目錄下的“查找和選擇”目錄.
Isnull是Python中檢驗空值的函數(shù)
#檢查數(shù)據(jù)空值
df.isnull()
#檢查特定列空值
df['price'].isnull()
5.查看唯一值
Excel中查看唯一值的方法是使用“條件格式”對唯一值進行顏色
標(biāo)記。
Python中使用unique函數(shù)查看唯一值革答。
#查看city列中的唯一值
df['city'].unique()
array(['Beijing ', 'SH', ' guangzhou ', 'Shenzhen', 'shanghai', '
BEIJING '], dtype=object)
6.查看數(shù)據(jù)表數(shù)值
Python中的Values函數(shù)用來查看數(shù)據(jù)表中的數(shù)值
#查看數(shù)據(jù)表的值
df.values
7.查看列名稱
Colums函數(shù)用來單獨查看數(shù)據(jù)表中的列名稱战坤。
#查看列名稱
df.columns
Index(['id', 'date', 'city', 'category', 'age', 'price'], dtype='
object')
8.查看前10行數(shù)據(jù)
Head函數(shù)用來查看數(shù)據(jù)表中的前N行數(shù)據(jù)
#查看前3行數(shù)據(jù)
df.head(3)
9.查看后10行數(shù)據(jù)
Tail行數(shù)與head函數(shù)相反,用來查看數(shù)據(jù)表中后N行的數(shù)據(jù)
#查看最后3行
df.tail(3)
數(shù)據(jù)表清洗
本章介紹對數(shù)據(jù)表中的問題進行清洗残拐,包括對空值途茫、大小寫問題、數(shù)據(jù)格式和重復(fù)值的處理溪食。
1.處理空值(刪除或填充)
Excel中可以通過“查找和替換”功能對空值進行處理
Python中處理空值的方法比較靈活囊卜,可以使用 Dropna函數(shù)用來刪除數(shù)據(jù)表中包含空值的數(shù)據(jù),也可以使用fillna函數(shù)對空值進行填充错沃。
#刪除數(shù)據(jù)表中含有空值的行
df.dropna(how='any')
也可以使用數(shù)字對空值進行填充
#使用數(shù)字0填充數(shù)據(jù)表中空值
df.fillna(value=0)
使用price列的均值來填充NA字段栅组,同樣使用fillna函數(shù),在要填充的數(shù)值中使用mean函數(shù)先計算price列當(dāng)前的均值枢析,然后使用這個均值對NA進行填充玉掸。
#使用price均值對NA進行填充
df['price'].fillna(df['price'].mean())
Out[8]:
0 1200.0
1 3299.5
2 2133.0
3 5433.0
4 3299.5
5 4432.0
Name: price, dtype: float64
2.清理空格
字符中的空格也是數(shù)據(jù)清洗中一個常見的問題
#清除city字段中的字符空格
df['city']=df['city'].map(str.strip)
3.大小寫轉(zhuǎn)換
在英文字段中,字母的大小寫不統(tǒng)一也是一個常見的問題登疗。
Excel中有UPPER排截,LOWER等函數(shù)嫌蚤,Python中也有同名函數(shù)用來解決
大小寫的問題。
#city列大小寫轉(zhuǎn)換
df['city']=df['city'].str.lower()
4.更改數(shù)據(jù)格式
Excel中通過“設(shè)置單元格格式”功能可以修改數(shù)據(jù)格式断傲。
Python中通過astype函數(shù)用來修改數(shù)據(jù)格式颗品。
#更改數(shù)據(jù)格式
df['price'].astype('int')
0 1200
1 3299
2 2133
3 5433
4 3299
5 4432
Name: price, dtype: int32
5.更改列名稱
Rename是更改列名稱的函數(shù)切油,我們將來數(shù)據(jù)表中的category列更改為category-size阐斜。
#更改列名稱
df.rename(columns={'category': 'category-size'})
6.刪除重復(fù)值
Excel的數(shù)據(jù)目錄下有“刪除重復(fù)項”的功能
Python中使用drop_duplicates函數(shù)刪除重復(fù)值
df['city']
0 beijing
1 sh
2 guangzhou
3 shenzhen
4 shanghai
5 beijing
Name: city, dtype: object
city列中beijing存在重復(fù)掰担,分別在第一位和最后一位
drop_duplicates()函數(shù)刪除重復(fù)值
#刪除后出現(xiàn)的重復(fù)值
df['city'].drop_duplicates()
0 beijing
1 sh
2 guangzhou
3 shenzhen
4 shanghai
Name: city, dtype: object
設(shè)置keep='last‘’參數(shù)后,與之前刪除重復(fù)值的結(jié)果相反垦垂,第一位
出現(xiàn)的beijing被刪除
#刪除先出現(xiàn)的重復(fù)值
df['city'].drop_duplicates(keep='last')
1 sh
2 guangzhou
3 shenzhen
4 shanghai
5 beijing
Name: city, dtype: objec
7.數(shù)值修改及替換
Excel中使用“查找和替換”功能就可以實現(xiàn)數(shù)值的替換
Python中使用replace函數(shù)實現(xiàn)數(shù)據(jù)替換
#數(shù)據(jù)替換
df['city'].replace('sh', 'shanghai')
0 beijing
1 shanghai
2 guangzhou
3 shenzhen
4 shanghai
5 beijing
Name: city, dtype: object
數(shù)據(jù)預(yù)處理
本章主要講的是數(shù)據(jù)的預(yù)處理宦搬,對清洗完的數(shù)據(jù)進行整理以便后期的統(tǒng)計和分析工作。主要包括數(shù)據(jù)表的合并劫拗,排序间校,數(shù)值分列,數(shù)據(jù)分組及標(biāo)記等工作页慷。
1.數(shù)據(jù)表合并
在Excel中沒有直接完成數(shù)據(jù)表合并的功能憔足,可以通過VLOOKUP函數(shù)分步實現(xiàn)。
在Python中可以通過merge函數(shù)一次性實現(xiàn)酒繁。
#建立df1數(shù)據(jù)表
df1=pd.DataFrame({"id":[1001,1002,1003,1004,1005,1006,1007,1008],
"gender":['male','female','male','female','male
','female','male','female'],
"pay":['Y','N','Y','Y','N','Y','N','Y',],
"m-point":[10,12,20,40,40,40,30,20]})
使用merge函數(shù)對兩個數(shù)據(jù)表進行合并滓彰,合并的方式為inner,將
兩個數(shù)據(jù)表中共有的數(shù)據(jù)匹配到一起生成新的數(shù)據(jù)表州袒。并命名為
df_inner揭绑。
#數(shù)據(jù)表匹配合并
df_inner=pd.merge(df,df1,how='inner')
合并的方式還有l(wèi)eft,right和outer方式
df_left=pd.merge(df,df1,how='left')
df_right=pd.merge(df,df1,how='right')
df_outer=pd.merge(df,df1,how='outer')
2.設(shè)置索引列
索引列可以進行數(shù)據(jù)提取郎哭,匯總他匪,數(shù)據(jù)篩選
#設(shè)置索引列
df_inner.set_index('id')
3.排序(按索引,按數(shù)值)
Excel中可以通過數(shù)據(jù)目錄下的排序按鈕直接對數(shù)據(jù)表進行排
序
Python中需要使用ort_values函數(shù)和sort_index函數(shù)完成排序
#按特定列的值排序
df_inner.sort_values(by=['age'])
Sort_index函數(shù)用來將數(shù)據(jù)表按索引列的值進行排序彰居。
#按索引列排序
df_inner.sort_index()
4.數(shù)據(jù)分組
Excel中可以通過VLOOKUP函數(shù)進行近似匹配來完成對數(shù)值的分組诚纸,或者使用“數(shù)據(jù)透視表”來完成分組
Python中使用Where函數(shù)用來對數(shù)據(jù)進行判斷和分組
#如果price列的值>3000,group列顯示high陈惰,否則顯示low
df_inner['group'] = np.where(df_inner['price'] > 3000,'high','low
')
還可以對多個字段的值進行判斷后對數(shù)據(jù)進行分組,下面的代碼中對city列等于beijing并且price列大于等于4000的數(shù)據(jù)標(biāo)記為1毕籽。
#對復(fù)合多個條件的數(shù)據(jù)進行分組標(biāo)記
df_inner.loc[(df_inner['city'] == 'beijing') & (df_inner['price']
>= 4000), 'sign']=1
5.數(shù)據(jù)分列
Excel中的數(shù)據(jù)目錄下提供“分列”功能抬闯。
在Python中使用split函數(shù)實現(xiàn)分列在數(shù)據(jù)表中category列中的數(shù)據(jù)包含有兩個信息,前面的數(shù)字為類別id关筒,后面的字母為size值溶握。中間以連字符進行連接。我們使用split函數(shù)對這個字段進行拆分蒸播,并將拆分后的數(shù)據(jù)表匹配回原數(shù)據(jù)表中睡榆。
#對category字段的值依次進行分列萍肆,并創(chuàng)建數(shù)據(jù)表,索引值為df_inner的索引列胀屿,列名稱為category和size
pd.DataFrame((x.split('-') for x in df_inner['category']),index=d
f_inner.index,columns=['category','size'])
#將完成分列后的數(shù)據(jù)表與原df_inner數(shù)據(jù)表進行匹配
df_inner=pd.merge(df_inner,split,right_index=True, left_index=Tru
e)
數(shù)據(jù)提取
1.按標(biāo)簽提取(loc)
#按索引提取單行的數(shù)值
df_inner.loc[3]
id 1004
date 2013-01-05 00:00:00
city shenzhen
category 110-C
age 32
price 5433
gender female
m-point 40
pay Y
group high
sign NaN
category_1 110
size C
Name: 3, dtype: object
使用冒號可以限定提取數(shù)據(jù)的范圍塘揣,冒號前面為開始的標(biāo)簽值后面為結(jié)束的標(biāo)簽值。
#按索引提取區(qū)域行數(shù)值
df_inner.loc[0:5]
Reset_index函數(shù)用于恢復(fù)索引宿崭,這里我們重新將date字段的日期
設(shè)置為數(shù)據(jù)表的索引亲铡,并按日期進行數(shù)據(jù)提取。
#重設(shè)索引
df_inner.reset_index()
#設(shè)置日期為索引
df_inner=df_inner.set_index('date')
#提取4日之前的所有數(shù)據(jù)
df_inner[:'2013-01-04']
2.按位置提取(iloc)
使用iloc函數(shù)按位置對數(shù)據(jù)表中的數(shù)據(jù)進行提取葡兑,這里冒號前后
的數(shù)字不再是索引的標(biāo)簽名稱奖蔓,而是數(shù)據(jù)所在的位置,從0開始讹堤。
#使用iloc按位置區(qū)域提取數(shù)據(jù)
df_inner.iloc[:3,:2]
iloc函數(shù)除了可以按區(qū)域提取數(shù)據(jù)吆鹤,還可以按位置逐條提取
#使用iloc按位置單獨提取數(shù)據(jù)
df_inner.iloc[[0,2,5],[4,5]]
前面方括號中的0,2,5表示數(shù)據(jù)所在行的位置,后面方括號中的數(shù)表示所在列的位置洲守。
3.按標(biāo)簽和位置提纫晌瘛(ix)
ix是loc和iloc的混合,既能按索引標(biāo)簽提取岖沛,也能按位置進行數(shù)
據(jù)提取.
#使用ix按索引標(biāo)簽和位置混合提取數(shù)據(jù)
df_inner.ix[:'2013-01-03',:4]
4.按條件提仁钍肌(區(qū)域和條件值)
使用loc和isin兩個函數(shù)配合使用,按指定條件對數(shù)據(jù)進行提取
#判斷city列的值是否為beijing
df_inner['city'].isin(['beijing'])
date
2013-01-02 True
2013-01-05 False
2013-01-07 True
2013-01-06 False
2013-01-03 False
2013-01-04 False
Name: city, dtype: bool
將isin函數(shù)嵌套到loc的數(shù)據(jù)提取函數(shù)中婴削,將判斷結(jié)果為Ture數(shù)據(jù)
提取出來廊镜。這里我們把判斷條件改為city值是否為beijing和shanghai。如果是就把這條數(shù)據(jù)提取出來唉俗。
#先判斷city列里是否包含beijing和shanghai嗤朴,然后將復(fù)合條件的數(shù)據(jù)提取出來。
df_inner.loc[df_inner['city'].isin(['beijing','shanghai'])]
數(shù)據(jù)篩選
按條件篩選(與虫溜、或雹姊、非)
Excel數(shù)據(jù)目錄下提供了“篩選”功能,用于對數(shù)據(jù)表按不同的條
件進行篩選衡楞。
Python中使用loc函數(shù)配合篩選條件來完成篩選功能吱雏。配合sum和count函數(shù)還能實現(xiàn)Excel中sumif和countif函數(shù)的功能。
使用“與”條件進行篩選瘾境,條件是年齡大于25歲歧杏,并且城市為
beijing。
#使用“與”條件進行篩選
df_inner.loc[(df_inner['age'] > 25) & (df_inner['city'] == 'beiji
ng'), ['id','city','age','category','gender']]/
#使用“或”條件篩選
df_inner.loc[(df_inner['age'] > 25) | (df_inner['city'] == 'beiji
ng'), ['id','city','age','category','gender']].sort(['age'])
#使用“非”條件進行篩選
df_inner.loc[(df_inner['city'] != 'beijing'), ['id','city','age',
'category','gender']].sort(['id'])
在前面的代碼后面增加city列迷守,并使用count函數(shù)進行計數(shù)犬绒。相當(dāng)于Excel中的countifs函數(shù)的功能
#對篩選后的數(shù)據(jù)按city列進行計數(shù)
df_inner.loc[(df_inner['city'] != 'beijing'), ['id','city','age',
'category','gender']].sort(['id']).city.count()
還有一種篩選的方式是用query函數(shù)
#使用query函數(shù)進行篩選
df_inner.query('city == ["beijing", "shanghai"]')
在前面的代碼后增加price字段和sum函數(shù)。對篩選后的price字段
進行求和兑凿,相當(dāng)于Excel中的sumifs函數(shù)的功能凯力。
#對篩選后的結(jié)果按price進行求和
df_inner.query('city == ["beijing", "shanghai"]').price.sum()
12230
數(shù)據(jù)匯總
Excel中使用分類匯總和數(shù)據(jù)透視可以按特定維度對數(shù)據(jù)進行匯總茵瘾,Python中使用的主要函數(shù)是groupby和pivot_table。
1.分類匯總
#對所有列進行計數(shù)匯總
df_inner.groupby('city').count()/
#對特定的ID列進行計數(shù)匯總
df_inner.groupby('city')['id'].count()
city
beijing 2
guangzhou 1
shanghai 2
shenzhen 1
Name: id, dtype: int64
#對兩個字段進行匯總計數(shù)
df_inner.groupby(['city','size'])['id'].count()
city size
beijing A 1
F 1
guangzhou A 1
shanghai A 1
B 1
shenzhen C 1
Name: id, dtype: int64
還可以對匯總后的數(shù)據(jù)同時按多個維度進行計算
#對city字段進行匯總并計算price的合計和均值咐鹤。
df_inner.groupby('city')['price'].agg([len,np.sum, np.mean])
2.數(shù)據(jù)透視
Python中通過pivot_table函數(shù)實現(xiàn)同樣的效果
#設(shè)定city為行字段拗秘,size為列字段,price為值字段慷暂。
分別計算price的數(shù)量和金額并且按行與列進行匯總聘殖。
pd.pivot_table(df_inner,index=["city"],values=["price"],columns=[
"size"],aggfunc=[len,np.sum],fill_value=0,margins=True)
數(shù)據(jù)統(tǒng)計
1.數(shù)據(jù)采樣
Excel的數(shù)據(jù)分析功能中提供了數(shù)據(jù)抽樣的功能
Python通過sample函數(shù)完成數(shù)據(jù)采樣
#簡單的數(shù)據(jù)采樣
df_inner.sample(n=3)
Weights參數(shù)是采樣的權(quán)重,通過設(shè)置不同的權(quán)重可以更改采樣的結(jié)果
#手動設(shè)置采樣權(quán)重
weights = [0, 0, 0, 0, 0.5, 0.5]
df_inner.sample(n=2, weights=weights)
Sample函數(shù)中參數(shù)replace行瑞,用來設(shè)置采樣后是否放回
#采樣后不放回
df_inner.sample(n=6, replace=False)
#采樣后放回
df_inner.sample(n=6, replace=True)
2.描述統(tǒng)計
Python中可以通過Describe對數(shù)據(jù)進行描述統(tǒng)計
#數(shù)據(jù)表描述性統(tǒng)計
df_inner.describe().round(2).T
3.相關(guān)分析
Python中則通過corr函數(shù)完成相關(guān)分析的操作奸腺,并返回相關(guān)系數(shù)。
#相關(guān)性分析
df_inner['price'].corr(df_inner['m-point'])
0.77466555617085264
#數(shù)據(jù)表相關(guān)性分析
df_inner.corr()
數(shù)據(jù)輸出
1.寫入Excel
#輸出到Excel格式
df_inner.to_Excel('Excel_to_Python.xlsx', sheet_name='bluewhale_c
c')
2.寫入csv
#輸出到CSV格式
df_inner.to_csv('Excel_to_Python.csv')
參考
王彥平《從Excel到Python:數(shù)據(jù)分析進階指南》
首發(fā)于微信公眾號:機器學(xué)習(xí)算法與Python實戰(zhàn)