首页 > 代码库 > 《利用python进行数据分析》读书笔记--第五章 pandas入门

《利用python进行数据分析》读书笔记--第五章 pandas入门

http://www.cnblogs.com/batteryhp/p/5006274.html

 

pandas是本书后续内容的首选库。pandas可以满足以下需求:

  • 具备按轴自动或显式数据对齐功能的数据结构。这可以防止许多由于数据未对齐以及来自不同数据源(索引方式不同)的数据而导致的常见错误。.
  • 集成时间序列功能
  • 既能处理时间序列数据也能处理非时间序列数据的数据结构
  • 数学运算和简约(比如对某个轴求和)可以根据不同的元数据(轴编号)执行
  • 灵活处理缺失数据
  • 合并及其他出现在常见数据库(例如基于SQL的)中的关系型运算

1、pandas数据结构介绍

两个数据结构:Series和DataFrame。Series是一种类似于以为NumPy数组的对象,它由一组数据(各种NumPy数据类型)和与之相关的一组数据标签(即索引)组成的。可以用index和values分别规定索引和值。如果不规定索引,会自动创建 0 到 N-1 索引。

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
from pandas import Series,DataFrame

#Series可以设置index,有点像字典,用index索引
obj = Series([1,2,3],index=[‘a‘,‘b‘,‘c‘])
#print obj[‘a‘]
#也就是说,可以用字典直接创建Series

dic = dict(key = [‘a‘,‘b‘,‘c‘],value = http://www.mamicode.com/[1,2,3])>

DataFrame是一种表格型结构,含有一组有序的列,每一列可以是不同的数据类型。既有行索引,又有列索引,可以被看做由Series组成的字典(使用共同的索引)。跟其他类似的数据结构(比如R中的data.frame),DataFrame面向行和列的操作基本是平衡的。其实,DataFrame中的数据是以一个或者多个二维块存放的(不是列表、字典或者其他)。

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
from pandas import Series,DataFrame

#构建DataFrame可以直接传入等长的列表或Series组成的字典
#不等长会产生错误
data = http://www.mamicode.com/{‘a‘:[1,2,3],>

另一种常见的结构是嵌套字典,即字典的字典,这样的结构会默认为外键为列,内列为行。

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
from pandas import Series,DataFrame
#内层字典的键值会被合并、排序以形成最终的索引
pop = {‘Nevada‘:{2001:2.4,2002:2.9},
       ‘Ohio‘:{2000:1.5,2001:1.7,2002:3.6}}
frame3 = DataFrame(pop)
#rint frame3
#Dataframe也有行和列有name属性,DataFrame有value属性
frame3.index.name = ‘year‘
frame3.columns.name = ‘state‘
print frame3
print frame3.values

下面列出了DataFrame构造函数能够接受的各种数据。

技术分享

索引对象

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
from pandas import Series,DataFrame
#pandas索引对象负责管理轴标签和其他元数据,构建Series和DataFrame时,所用到的任何数组或其他序列的标签都被转换为Index
obj = Series(range(3),index = [‘a‘,‘b‘,‘c‘])
index = obj.index
#print index
#索引对象是无法修改的,这非常重要,因为这样才会使得Index对象在多个数据结构之间安全共享
index1 = pd.Index(np.arange(3))
obj2 = Series([1.5,-2.5,0],index = index1)
print obj2.index is index1

#除了长得像数组,Index的功能也类似一个固定大小的集合
print ‘Ohio‘ in frame3.columns
print 2003 in frame3.index

pandas中的Index是一个类,pandas中主要的Index对象(什么时候用到)。

技术分享

下面是Index的方法与属性,值得注意的是:index并不是数组。

技术分享

2、基本功能

下面介绍基本的Series 和 DataFrame 数据处理手段。首先是索引:

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame

#Series有一个reindex函数,可以将索引重排,以致元素顺序发生变化

obj = Series([1,2,3,4],index=[‘a‘,‘b‘,‘c‘,‘d‘])
#注意这里的reindex并不改变obj的值,得到的是一个“副本”
#fill_value 显然是填充空的index的值
#print obj.reindex([‘a‘,‘c‘,‘d‘,‘b‘,‘e‘],fill_value = http://www.mamicode.com/0)>

关于ix,是DataFrame的一个方法,http://pandas.pydata.org/pandas-docs/version/0.17.0/generated/pandas.DataFrame.ix.html。

技术分享

技术分享

丢弃指定轴上的项

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame
#drop函数可以丢弃轴上的列、行值
obj = Series(np.arange(3.),index = [‘a‘,‘b‘,‘c‘])
#原Series并不丢弃
obj.drop(‘b‘)
#print obj
#注意下面,行可以随意丢弃,列需要加axis = 1
print frame.drop([‘a‘])
print frame.drop([‘Ohio‘],axis = 1)

下面说索引、选取和过滤

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame

obj = Series([1,2,3,4],index=[‘a‘,‘b‘,‘c‘,‘d‘])
frame = DataFrame(np.arange(9).reshape((3,3)),index = [‘a‘,‘c‘,‘d‘],columns = [‘Ohio‘,‘Texas‘,‘California‘])

#Series切片和索引
#print obj[obj < 2]
#注意:利用标签的切片与python的切片不同,两端都是包含的(有道理)
print obj[‘b‘:‘c‘]
#对于DataFrame,列可以直接用名称
print frame[‘Ohio‘]
#特殊情况:通过切片和bool型索引,得到的是行(有道理)
print frame[:2]
print frame[frame[‘Ohio‘] != 0]
#下面的方式是对frame所有元素都适用,不是行或者列,下面的得到的是numpy.ndarray类型的数据
print frame[frame < 5],type(frame[frame < 5])
frame[frame < 5] = 0
print frame

#对于DataFrame上的标签索引,用ix进行
print frame.ix[[‘a‘,‘d‘],[‘Ohio‘,‘Texas‘]]
print frame.ix[2] #注意这里默认取行
#注意下面默认取行
print frame.ix[frame.Ohio > 0]
#注意下面的逗号后面是列标
print frame.ix[frame.Ohio > 0,:2]

下面是常用的索引选项:

技术分享

技术分享

算术运算和数据对齐

#pandas 有一个重要的功能就是能够根据索引自动对齐,其中索引不重合的部分值为NaN
s1 = Series([1,2,3],[‘a‘,‘b‘,‘c‘])
s2 = Series([4,5,6],[‘b‘,‘c‘,‘d‘])
#print s1 + s2
df1 = DataFrame(np.arange(12.).reshape(3,4),columns=list(‘abcd‘))
df2 = DataFrame(np.arange(20.).reshape(4,5),columns=list(‘abcde‘))
#print df1 + df2
#使用add方法,并传入填充值,注意下面的fill_value函数是先对应填充再进行加和,而不是加和得到NaN之后再填充
#print df1.add(df2,fill_value = http://www.mamicode.com/1000)>

除了add之外,还有其他的方法:

技术分享

DataFrame和Series之间的运算

#下面看一下DataFrame和Series之间的计算过程
arr = DataFrame(np.arange(12.).reshape((3,4)),columns = list(‘abcd‘))
#下面的结果标明,就是按行分别相减即可,叫做 broadcasting
#注意:默认情况下,DataFrame和Series的计算会将Series的索引匹配到DataFrame的列,然后进行计算,再沿着行一直向下广播
#注意:下面的式子中,如果写arr - arr[0]是错的,因为只有标签索引函数ix后面加数字才表示行
print arr - arr.ix[0]
Series2 = Series(range(3),index = list(‘cdf‘))
#按照规则,在不匹配的列会形成NaN值
print arr + Series2
#如果想匹配行且在列上广播,需要用到算术运算方法
Series3 = arr[‘d‘]
#axis就是希望匹配的轴
print arr.sub(Series3,axis = 0)

下面是函数应用和映射

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame

#NumPy的元素级数组方法也适用于pandas对象
frame = DataFrame(np.random.randn(4,3),columns = list(‘abc‘),index = [‘Ut‘,‘Oh‘,‘Te‘,‘Or‘])
print frame
#下面是求绝对值:
#print np.abs(frame)
#另一种常见的做法是:将一个函数应用到行或者列上,用apply方法,与R语言类似
fun = lambda x:x.max() - x.min()
#默认是应用在每一列上
print frame.apply(fun)
#下面是应用在列上
print frame.apply(fun,axis = 1)
#很多统计函数根本不用apply,直接调用方法就可以了
print frame.sum()
#除了标量值之外,apply函数后面还可以接返回多个值组成的的Series的函数,有没有很漂亮?
def f(x):
    return Series([x.min(),x.max()],index = [‘min‘,‘max‘])
#print frame.apply(f)
#元素级的python函数也是可以用的,但是要使用applymap函数
format = lambda x: ‘%.2f‘ % x
print frame.applymap(format)
#之所以要用applymap是因为Series有一个应用于元素级函数的map方法??
#这里的map很有用
print frame[‘b‘].map(format)

排序与排名

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame
#用sort_index函数对行、列的索引进行排序
obj = Series(range(4),index = [‘d‘,‘a‘,‘b‘,‘c‘])
print obj.sort_index()

frame = DataFrame(np.arange(8).reshape((2,4)),index = [‘three‘,‘one‘],columns = [‘d‘,‘a‘,‘b‘,‘c‘])
#默认是对行 “索引” 进行排序,如果对列 “索引” 进行排序,axis = 1 即可
print frame.sort_index()
print frame.sort_index(axis = 1)
print frame.sort_index(axis = 1,ascending = False)

#如果对值进行排序,用的是order函数,注意所有的缺失值会放到最后(如果有的话)
print obj.order()
#numpy中的sort也可以用来排序
print np.sort(obj)
#如果相对DataFrame的值进行排序,函数还是sort_index,只不过后面需要加一个参数by
frame = DataFrame({‘b‘:[4,7,-3,2],‘a‘:[0,1,0,1]})
print frame.sort_index(by = [‘a‘,‘b‘])

#rank函数返回从小到大排序的下标,对于平级的数,rank是通过“为各组分配一个平均排名”的方式破坏评级关系
#下标从1开始
obj = Series([7,-5,7,4,2,0,4])
print obj.rank()
#而numpy中的argsort函数比较奇怪,返回的是把数据进行排序之后,按照值得顺序对应的下标,下标从0开始
print np.argsort(obj)
 #打印结果为:1,5,4,3,6,0,2 按照这个下标顺序恰好可以得到从小打到的值,见下面
print obj[np.argsort(obj)]
#rank函数中有一个method选项,用来规定下标的方式

print obj.rank(method = ‘first‘,ascending=False)
print obj.rank(method = ‘max‘,ascending=False)
print obj.rank(method = ‘min‘,ascending=False)

#对于DataFrame,rank函数默认把每一列排好并返回坐标
print frame.rank()
print frame.rank(axis = 1)

带有重复值的轴索引

#-*- encoding:utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from pandas import Series,DataFrame

#虽然pandas的很多函数(如reindex)要求标签唯一,但是并不具有强制性
obj = Series(range(5),index = list(‘aabbc‘))
print obj
#索引是否唯一用is_unique看是否唯一
print obj.index.is_unique
#对于重复值的索引,选取的话返回一个Series,唯一的索引返回一个标量
print obj[‘a‘]
#对于DataFrame也是如此
df = DataFrame(np.random.randn(4,3),index = list(‘aabb‘))
print df
print df.ix[‘b‘]
#####自己导入数据的时候数据处理之前可以做一下index唯一性等,自己创建DataFrame注意不能这样

3、汇总和计算描述统计

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time

#pandas 对象拥有一组常用的数学和统计方法,大部分属于简约统计,用于从Series中提取一个值,或者   从DataFrame中提取一列或者一行Series
#注意:与NumPy数组相比,这些函数都是基于没有缺失数据的建设构建的,也就是说:这些函数会自动忽略缺失值。
df = DataFrame([[1.4,np.nan],[7.1,-4.5],[np.nan,np.nan],[0.75,-1.3]],index = list(‘abcd‘),columns=[‘one‘,‘two‘])
print df.sum()
print df.sum(axis = 1)
#下面是一些函数,idxmin 和 idmax 返回的是达到最小或者最大的索引
print df.idxmin()
print df.idxmin(axis=1)
#关于累积型的函数
print df.cumsum()
#describe函数,与R语言中的describe函数基本相同
print df.describe()
#对于非数值型的数据,看看下面的结果

obj = Series([‘c‘,‘a‘,‘a‘,‘b‘,‘d‘] * 4)
print obj.describe()
‘‘‘
结果为:
count     20
unique     4
top        a
freq       8
其中,freq是指字母出现的最高频率
‘‘‘

技术分享

技术分享

技术分享

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time

#下面看一下cummin函数
#注意:这里的cummin函数是截止到目前为止的最小值,而不是加和以后的最小值
frame = DataFrame([[1,2,3,4],[5,6,7,8],[-10,11,12,-13]],index = list(‘abc‘),columns = [‘one‘,‘two‘,‘three‘,‘four‘])
print frame.cummin()
print frame
>>>
   one  two  three  four
a    1    2      3     4
b    1    2      3     4
c  -10    2      3   -13
   one  two  three  four
a    1    2      3     4
b    5    6      7     8
c  -10   11     12   -13

相关系数与协方差

有些汇总统计(如相关系数和协方差)是通过参数对计算出来的。这一节数据得不到?上不去网。

唯一值、值计数以及成员资格

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt

obj = Series([‘a‘,‘a‘,‘b‘,‘f‘,‘e‘])
uniques = obj.unique()
uniques.sort() #记住这是就地排序
#print uniques
#下面进行计数统计,注意得到的是按照出现的频率降序排列
#print obj.value_counts()
#value_counts还是一个顶级的pandas方法。可用于任何是数组或者序列
#print obj.values
#print pd.value_counts(obj.values,sort = False)
#最后是isin 判断矢量化集合的成员资格,可用于选取Series中或DataF列中的子集
mask = obj.isin([‘b‘,‘c‘])
print mask
print obj[mask]

data = http://www.mamicode.com/DataFrame({‘Qu1‘:[1,3,4,3,4],>

技术分享

上面这几个函数是真的非常实用!

4、处理缺失数据

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time
from numpy import nan as NA

#pandas本来就被设计成自动忽略了缺失值、
#nan None 都看做缺失值
str_data = http://www.mamicode.com/Series([‘a‘,np.nan,‘b‘,‘c‘])>
>>>
0     True
1     True
2    False
3    False
0    False
1    False
2     True
3     True
#NumPy的数据类型中缺少真正的NA数据类型或位模式??
 

技术分享

滤除缺失数据

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time
from numpy import nan as NA

data = http://www.mamicode.com/Series([1,NA,3.5,7,NA])>
#有一个函数 reset_index 这个函数(方法?)可以重新设置index,其中drop = True选项会丢弃原来的索引而设置新的从0开始的索引,这个方法只对DataFrame有用貌似。
print data.dropna()
#下面的结果一样
print data[data.notnull()]
data1 = DataFrame([[1,2,3],[NA,2.3,4],[NA,NA,NA]])
#注意:由于DataFrame的设定,只要有NA的行就会舍弃
print data1.dropna()
#传入how = ‘all‘ 则丢掉全为NA的行,这里的 how 的起名真的有点随心所欲了,哈哈
print data1.dropna(how = ‘all‘)
#丢弃列
print data1.dropna(how = ‘all‘,axis = 1)
#还有一个参数,thresh
data2 = DataFrame(np.random.randn(7,3))
data2.ix[:4,1] = NA
data2.ix[:2,2] = NA
#print data2
#这里的thresh函数是选取最少non-NA值个数的行选出来
print data2.dropna(thresh = 2)
print data2.dropna(thresh = 4,axis = 1)

填充缺失数据

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time
from numpy import nan as NA

#主要用fillna方法填充NA处的值
data2 = DataFrame(np.random.randn(7,3))
data2.ix[:4,1] = NA
data2.ix[:2,2] = NA
#fillna返回一个新对象,inplace = True 可以就地填充
print data2.fillna(0)
#print data2.fillna(0,inplace = True)
#print data2
#为不同的列填充要用到字典
print data2.fillna({1:0.5,3:-1})
#对reindex有效的的那些差值方法也可适用于fillna,请向上看,或者搜索 reindex 即可
df = DataFrame(np.random.randn(6,3))
df.ix[2:,1] = NA
df.ix[4:,2] = NA
print df.fillna(method = ‘ffill‘,limit = 2)
#只要稍微动动脑子,我们就可以知道向NA处可以填充均值等其他数
data = http://www.mamicode.com/Series([1.2,NA,4,NA])>

fillna的参数如下:

技术分享

技术分享

5、层次化索引

层次化索引(hierarchical index)是pandas的重要功能,这能使在一个轴上拥有两个以上的索引级别。抽象点说,它能使你以低维度形式处理高维度。

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time

data = http://www.mamicode.com/Series(np.random.randn(10),index=[[‘a‘,‘a‘,‘a‘,‘b‘,‘b‘,‘b‘,‘c‘,‘c‘,‘d‘,‘d‘],[1,2,3,1,2,3,1,2,2,3]])>
#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import time
#人们经常想将DataFrame的一个或者多个列当作行索引来用,或者可能需要将行索引变成DataFrame的列
frame = DataFrame({‘a‘:range(7),‘b‘:range(7,0,-1),‘c‘:[‘one‘,‘one‘,‘one‘,‘two‘,‘two‘,‘two‘,‘two‘],‘d‘:[0,1,2,0,1,2,3]})
print frame
#DataFrame中的set_index函数会将其一个或者多个列转换为行索引
frame2 = frame.set_index([‘c‘,‘d‘])
print frame2  #其实就是利用第3、4列进行一次分类汇总
frame3 = frame.set_index([‘c‘,‘d‘],drop = False)
#与set_index相反的是reset_index函数
print frame2.reset_index()
#下面进行一次测试
frame4 = DataFrame([[0,7],[1,6],[2,5],[3,4],[4,3],[5,2],[6,1]],index = [[‘one‘,‘one‘,‘one‘,‘two‘,‘two‘,‘two‘,‘two‘],[0,1,2,0,1,2,3]],columns=[‘a‘,‘b‘])
frame4.index.names = [‘c‘,‘d‘]
print frame4
print frame4.reset_index().sort_index(axis = 1)

其他有关pandas的话题

#-*- encoding:utf-8 -*-
import numpy as np
import os
import pandas as pd
from pandas import Series,DataFrame
import matplotlib.pyplot as plt
import pandas.io.data  as web
#这里说的是一些蛋疼的问题:整数索引和整数标签
ser = Series(np.arange(3.))
#print ser[-1]  #报错,因为整数索引的歧义性
ser2 = Series(np.arange(3.),index = [‘a‘,‘b‘,‘c‘])
print ser2[-1] #正确
#ix函数总是面向标签的
print ser.ix[:1]
#如果需要可靠的、不考虑索引类型的、基于位置的索引,可以使用Series的iget_value方法,Dataframe的irow 和 icol方法
ser3 = Series(range(3),index= [-5,1,3])
print ser3.iget_value(2)
frame = DataFrame(np.arange(6).reshape(3,2),index = [2,0,1])
print frame.irow(0)

#pandas 有一个Panel数据结构(不是主要内容),可以看作是三维的DataFrame。pandas中的多维数据可以利用多层索引进行处理
#可以利用DataFrame对象组成的字典或者一个三维ndarray来创建Panel对象
pdata = http://www.mamicode.com/pd.Panel(dict((stk,web.get_data_yahoo(stk,‘1/1/2009‘,‘6/1/2012‘)) for stk in [‘AAPL‘,‘GOOG‘,‘MSFT‘,‘DELL‘]))>

《利用python进行数据分析》读书笔记--第五章 pandas入门