Python使用Pandas库常见操作详解

本文实例讲述了python使用pandas库常见操作。分享给大家供大家参考，具体如下：

1、概述

pandas 是python的核心数据分析支持库，提供了快速、灵活、明确的数据结构，旨在简单、直观地处理关系型、标记型数据。pandas常用于处理带行列标签的矩阵数据、与 sql 或 excel 表类似的表格数据，应用于金融、统计、社会科学、工程等领域里的数据整理与清洗、数据分析与建模、数据可视化与制表等工作。

数据类型：pandas 不改变原始的输入数据，而是复制数据生成新的对象，有普通对象构成的一维数组成为series，由series构成的二维数组表称为dataframe，其行被称为index，列为colum。

安装：如果使用anaconda集成环境则会自动安装numpy、scipy、pandas等数据科学包，也可以通过python包管理工具安装pandas：

pip install pandas

2、数据对象的创建

通过series()函数包裹一维数组可以创建series对象，其中数组的元素可以是各种类型。

通过dataframe()函数包裹二维数组可以创建一个dataframe对象，可以通过参数index、columns指定行标签和列标签。也可以通过python的字典类型初始化dataframe，其键名默认为列标签

import pandas as pd
import numpy as np
 
# 通过一维数组初始化series
s = pd.series([1, 2.0, np.nan, 'test'])
print(s)
 
# 通过二维数组初始化dataframe
arr = np.random.randn(6, 4)
arr_df = pd.dataframe(arr, index=np.arange(1, 7), columns=list('abcd'))
print(arr_df)
# 通过字典dict初始化dataframe
dic = {'a': 1.,
    'b': pd.timestamp('20130102'),
    'c': pd.series(1, index=list(range(4)), dtype='float32'),
    'd': np.array([3] * 4, dtype='int32'),
    'e': pd.categorical(["test", "train", "test", "train"])
    }
dic_df = pd.dataframe(dic)
print(dic_df)

其运行结果如下：

# series数据
0    1
1    2
2   nan
3  test
dtype: object
# 二维数组的dataframe
     a     b     c     d
1 -0.085417 -0.816502 1.495134 -0.277742
2 1.657144 -0.203346 0.631930 -1.182239
3 -2.303923 -0.535696 1.315379 0.129682
4 0.133198 -0.239664 -2.004494 0.119965
5 -1.454717 2.114255 -0.538678 -0.580361
6 -0.759183 0.141554 -0.243270 2.840325
# dict字典dataframe
   a     b  c d   e
0 1.0 2013-01-02 1.0 3  test
1 1.0 2013-01-02 1.0 3 train
2 1.0 2013-01-02 1.0 3  test
3 1.0 2013-01-02 1.0 3 train

3、查看数据

函数head(n)可以查看dataframe前n行的数据，tail(n)查看倒数n行的数据

index()查看dataframe的行标签，columns显示列标签

describe()按列显示数据的统计信息，包括计数、均值、方差、最小最大值等。

函数mean()显示所有列的均值，mean(1)显示所有行的均值

sum()求所有列的均值，sum(1)求所有行的均值

dataframe有一个empty属性用于判断是否为空，若为空则返回true

arr = np.random.randn(6, 4)
df = pd.dataframe(arr, index=np.arange(1, 7), columns=list('abcd'))
print(df.head(3))
print(df.index)
print(df.describe())

结果如下

# 查看前三行数据
a     b     c     d
1 3.260449 -0.619396 0.070877 1.586914
2 -0.529708 0.071917 -1.919316 1.845727
3 -1.005765 2.176579 -0.323483 -1.295067
# 查看行标签
int64index([1, 2, 3, 4, 5, 6], dtype='int64')
# 查看统计信息
       a     b     c     d
count 6.000000 6.000000 6.000000 6.000000
mean -0.184606 -0.487184 0.079433 0.855810
std  1.721394 1.800460 1.379498 1.128764
min  -1.443635 -3.091446 -1.919316 -1.295067
25%  -0.967105 -1.430192 -0.281188 0.778729
50%  -0.694488 -0.273739 -0.041713 1.150944
75%  -0.531744 0.197755 0.355731 1.508475
max  3.260449 2.176579 2.352142 1.845727

4、数据的选择

可以直接通过dataframe对象选取列或者行，

# 选取一个列a，等价于df['a']
print(df.a)
# 选取第1到第3行，行下标从0开始
print(df[1:3])
 
'''
# 标签为a的那一列
1  0.644427
2  0.643149
3  1.374668
4  -0.154465
5  -0.338085
6  -1.989284
name: a, dtype: float64
# 第1~3行
     a     b     c     d
2 0.643149 1.769001 -0.166028 -0.036854
3 1.374668 -0.118593 -0.202222 0.308353
'''

通过loc[]方法可以通过标签对dataframe的一行、一列、几行几列或者是某个具体的值进行选择

# 取出行标签为2的那一行
print(df.loc[2])
# 取出行标签为1~3，列标签为'a','b'的内容
print(df.loc[1:3, ['a', 'b']])
# 获取行标签为1，列标签为'a'的具体值，等价于df.at[1,'a']
print(df.loc[1, 'a'])
 
'''
# 标签为2的一行
a  0.681469
b  -0.053046
c  -1.384877
d  -0.447700
name: 2, dtype: float64
# 标签为1~3，列标签为'a','b'的内容
     a     b
1 0.710907 -0.950896
2 0.681469 -0.053046
3 0.781981 0.123072
# 行标签为1，列标签为'a'的具体值
0.7109074858947351
'''

除了通过行列标签来进行取值以外，还可以通过行列的数组的位置进行取值，其方法名为iloc[]

# 取出第一行，行下标从0开始
print(df.iloc[0])
# 显示第1，2，4行的第0，2列
print(df.iloc[[1, 2, 4], [0, 2]])
# 显示第1行第1列的具体值，等价于df.iat[1,1]
print(df.iloc[1, 1])

还可以在选择时对数据进行过滤

# 输出a那一列大于0的所有行
print(df[df.a > 0])
df['e'] = ['one', 'one', 'two', 'three', 'four', 'three']
# 输出e那一列存在two、four的所有行
print(df[df['e'].isin(['two', 'four'])])
 
'''
     a     b     c     d
3 0.168998 -0.732362 -0.098542 0.413128
5 0.513677 -0.163231 -0.098037 -0.606693

     a     b     c     d   e
3 0.168998 -0.732362 -0.098542 0.413128  two
5 0.513677 -0.163231 -0.098037 -0.606693 four
'''

5、操作数据

通过insert()方法可以实现在指定位置插入一列，也可以直接将一个数组赋值给dataframe，这将默认添加到最后一列

可以通过之前的选择方法loc、iloc找到指定的行列，然后直接赋值，如果该位置存在数据则会修改，否则添加

通过drop()方法删除指定的数据，index属性指定删除的行，columns指定删除的列，inplace属性是否在原数据集上操作，默认为false，此时需要一个变量来接收删除后的结果

df = pd.dataframe(data = [['lisa','f',22],['joy','f',22],['tom','m','21']],
         index = [1,2,3],columns = ['name','sex','age'])
citys = ['ny','zz','xy']
#在第0列，加上column名称为city，值为citys的数值。
df.insert(0,'city',citys)
jobs = ['student','ai','teacher']
# 默认在df最后一列加上column名称为job，值为jobs的数据。
df['job'] = jobs
# 若df中没有index为“4”的这一行的话，则添加，否则修改
df.loc[4] = ['zz', 'mason', 'm', 24, 'engineer']
print(df)
# 删除行标签为1的行
dp=df.drop(index=1)
print(dp)
# 在原数据集上删除列标签为sex的列
df.drop(columns=['sex'],inplace=true)
print(df)

结果如下：

# 添加后的数据
 city  name sex age    job
1  ny  lisa  f 22  student
2  zz  joy  f 22    ai
3  xy  tom  m 21  teacher
4  zz mason  m 24 engineer
# 删除第一行
 city  name sex age    job
2  zz  joy  f 22    ai
3  xy  tom  m 21  teacher
4  zz mason  m 24 engineer
# 删除sex列
 city  name age    job
1  ny  lisa 22  student
2  zz  joy 22    ai
3  xy  tom 21  teacher
4  zz mason 24 engineer

对dataframe进行转置操作，调用.t

sort_index(axis=1, ascending=false)对数据进行排序，axis=0代表按行标签排序，axis=1代表按列标签排序

sort_values(by='a')按某一列的值对数据进行排序，这里是按列标签为a的

apply()函数对dataframe的每一行应用函数

print(df.t)
si=df.sort_index(axis=1, ascending=false)
print(si)
sv=df.sort_values(by='a')
print(sv)
# 应用匿名函数，用每一列最大值减去最小值
df.apply(lambda x: x.max() - x.min())
print(df)
'''
# 数据转置
     1     2     3     4     5     6
a -1.176180 -1.301768 0.907088 -1.528101 1.098978 -1.280193
b -0.461954 -0.749642 1.169118 -0.297765 0.531088 -0.999842
c -1.715094 -0.512856 0.511861 -0.247240 1.696772 -0.902995
d 1.336999 0.209091 2.254337 0.649625 -0.049886 -1.514815
# 按列标签倒序
    d     c     b     a
1 1.336999 -1.715094 -0.461954 -1.176180
2 0.209091 -0.512856 -0.749642 -1.301768
3 2.254337 0.511861 1.169118 0.907088
4 0.649625 -0.247240 -0.297765 -1.528101
5 -0.049886 1.696772 0.531088 1.098978
6 -1.514815 -0.902995 -0.999842 -1.280193
# 按列a的值递增对行排序
    a     b     c     d
4 -1.528101 -0.297765 -0.247240 0.649625
2 -1.301768 -0.749642 -0.512856 0.209091
6 -1.280193 -0.999842 -0.902995 -1.514815
1 -1.176180 -0.461954 -1.715094 1.336999
3 0.907088 1.169118 0.511861 2.254337
5 1.098978 0.531088 1.696772 -0.049886
# 函数的应用 
a  2.073961
b  2.671590
c  1.785291
d  0.000000
f  4.000000
dtype: float64
'''

panda的concat函数可以将两个相同类型的dataframe在行的维度上进行拼接

merge()函数可以将不同dataframe按列拼接

append()函数可以在dataframe的结尾追加

# 将第一行和最后一行拼接
print(pd.concat([df[:1], df[-2:-1]]))
# 将第4行追加到结尾
print(df.append(df.iloc[3]))
# 将两个dataframe按列拼接
df1 = pd.dataframe({'row1': ['foo', 'bar'], 'row2': [1, 2]})
df2 = pd.dataframe({'row1': ['foo', 'bar'], 'row3': [4, 5]})
print(pd.merge(df1, df2))
 
'''
# 按行拼接
     a     b     c     d
1 -0.527221 -0.754650 -2.385270 -2.569586
5 0.054059 1.443911 -0.240856 -1.501045
# 追加
     a     b     c     d
1 -0.527221 -0.754650 -2.385270 -2.569586
2 2.123332 -0.013431 -0.574359 -0.548838
3 -0.244057 -0.267805 1.089026 -0.022174
4 -0.789228 1.171906 0.526318 0.046655
5 0.054059 1.443911 -0.240856 -1.501045
6 0.756844 0.623305 -0.597299 0.034326
4 -0.789228 1.171906 0.526318 0.046655
# 按列拼接
 row1 row2 row3
0 foo   1   4
1 bar   2   5
'''

groupby函数可以数据按列进行分组，分组后的结果可以使用for循环进行迭代，迭代中每个分组是一个(index,dataframe)元组，可以对其中的dataframe作进一步操作。

stack()可以将多列的数据压缩为两列显示

df = pd.dataframe({'a': ['foo', 'bar', 'foo', 'bar'],
          'b': ['one', 'two', 'one', 'three'],
          'c': np.random.randn(4),
          'd': np.random.randn(4)})
# 按a、b两列进行分组
dg=df.groupby(['a', 'b'])
for (index,df) in dg:
    print(df)
# 压缩
print(df.stack())
 
'''
# 按列分组
   a   b     c     d
3 bar three 0.802027 1.338614
   a  b     c     d
1 bar two -0.567295 0.608978
   a  b    c     d
0 foo one -0.17592 -0.191991
2 foo one -0.72258 0.711312
# 压缩为两列
0 a     foo
  b     one
  c   0.622471
  d   0.10633
1 a     bar
  b     two
  c   0.065516
  d  -0.844223
2 a     foo
  b     one
  c  0.0013226
  d   -1.3328
3 a     bar
  b    three
  c  -0.678077
  d   0.785117
dtype: object
'''

pandas主要使用值np.nan来表示缺失的数据。可以使用dropna(how='any')方法来删除所有存在空值的行，dropna(axis=1)删除存在空值的列。fillna(value=x)用指定值x填充所有的空值。

6、其他

通过pandas可以便捷地从其他格式文件进行转换

# 将dataframe写入csv文件
df.to_csv('foo.csv')
# 从csv文件读数据
df = pd.read_csv('foo.csv')
# excel文件的读写
df = pd.read_excel('foo.xlsx', 'sheet1', index_col=none, na_values=['na'])
df.to_excel('foo.xlsx', sheet_name='sheet1')

pandas提供了便捷的时间维度生成函数date_range()，第一个参数是起始时间，periods=生成的数量，freq=时间间隔，默认以天为单位

# 从2019年1月1日开始，以秒为单位，生成五个时间
rng = pd.date_range('1/1/2019', periods=5, freq='s')
ts = pd.series(np.random.randint(0, 500, len(rng)), index=rng)
print(ts)
 
'''
2019-01-01 00:00:01  161
2019-01-01 00:00:02  214
2019-01-01 00:00:03  110
2019-01-01 00:00:04  265
freq: s, dtype: int32
'''

pandas结合matplot可以便捷地进行数据绘图

ts = pd.series(np.random.randint(0, 500, len(rng)), index=rng)
# 将数据追加到一个数组统一显示
ts=ts.cumsum()
# 调用matplot绘制图
ts.plot()