购买
下载掌阅APP,畅读海量书库
立即打开
畅读海量书库
扫码下载掌阅APP

第3章
Pandas数据提取

3.1 Pandas简介

3.1.1 Pandas数据分析

在Python中存在大量处理Excel的第三方库。在早期,用户可以通过xlrd库读取xls、xlsx文件,利用xlwt库写入xls文件,或者利用xlutils修改xls文件。这些库都有一个共同的特点:无法批量处理Excel文件。随着更多第三方库的加入,用户如果需要更高效地进行批量文件处理及数据分析,则可以采用Pandas、openpyxl及xlwings结合应用,实现远比Excel中VBA更为强大的功能。例如用户可以利用Pandas进行数据分析、利用Matplotlib或Seaborn生成数据图表、利用xlwings进行文件的批量格式处理及文件的读存,实现更高层级的BI(商业智能)。

本章主要介绍的是使用Pandas读取工作簿。对于刚接触Pandas的用户可能会有这样的疑问:什么是Pandas?Pandas能干什么?Pandas数据分析或Python数据分析到底是怎么一回事?简单来讲:Pandas是Python的一个数据分析包,Pandas的名称来自面板数据(Panel Data)和Python数据分析(Data Analysis)。Pandas的开发者是Wes McKinney(韦斯·麦金尼),该软件是他当时在纽约的AQR Capital Management(AQR资本投资管理公司)工作时,因为对Excel及R语言的数据清洗与汇总统计都感到不满意后而决定自己开发一个基于Python的数据分析软件。该软件于2008年完成了第1版,并于2009年12月向公众发布。可以这样说,Pandas是Python数据分析的利器。正是因为Pandas的加入才让Python在数据分析领域有了一席之地。Pandas在数据的增、删、改、查及时间序列等方面,功能十分强大。

Pandas中主要有Series和DataFrame两种数据结构。Series:一种类似于一维数组的对象,它是由一组数据及一组与之相关的数据标签(索引)组成的。DataFrame:一张表格型的数据结构,类似Excel、SQL表格等二维数据结构。含有一组有序的列,每列可以是不同的值类型(数值、字符串、布尔值等),DataFrame既有行索引,也有列索引,可以被看作由Series组成的字典。

注意: Series首字母必须大写,DataFrame采用的是驼峰式写法,D与F必须大写。

Pandas是Python的第三方库。按照常规约定,在使用Pandas之前,必须导入Pandas包,并以pd作为其别名,命令如下:

    import pandas as pd

如果在Pandas的使用过程中同时用到NumPy,则需同时导入NumPy包,并以np作为别名,命令如下:

    import numpy as np
    import pandas as pd

3.1.2 SMED与数据分析

SMED(Single Minute Exchange of Die)是“六十秒即时换模”的简称,是精益生产中的一个术语。在以往的生产制造过程中,产品更换模具及生产换线调整过程中总是要浪费大量的时间,从而造成生产的停顿与等待,当现场管理水平跟不上时,以往的做法总会采用批量生产(Mass Production)方式来减少更换模具或生产换线的时间,其最终的结果是“停顿的时间减少了,但库存的数量却猛增了”,这种生产方式显然无法应对弹性生产的需求,直至SMED(快速换模)得以推广后才真正将生产停顿及降低库存这两个问题一并得以解决,是对传统生产现场管理的革命性创新。

SMED的技术可以概括为“区分内作业与外作业,将内作业转换为外作业;减少外作业时间,减少内作业时间”。如果将此技术转换到数据清洗及数据分析场合中,则可以如下理解。

(1)区分规范数据与非规范数据、无效数据及冗余数据。

(2)在数据分析之前将不规范的数据转换为规范数据。

(3)在数据分析之时减少冗余,仅导入所需的数据。

(4)优化数据分析的代码,利用更快的工具、更好的方法快速处理数据,如图3-1所示。

图3-1 SMED使用流程

Python的代码一向以“优雅、简洁、高效”著称,Python的数据分析利器Pandas的代码同样以“灵活、高效、功能强大”著称,与SMED的理念高度吻合,以至于用户一提起Python数据分析首先想到的就是Pandas。

另外,之所以倡导在数据分析前将内作业转换为外作业是因为“在数据规范度较差的企业,数据分析师常常将80%以上的时间用于数据清洗,而仅将不到20%的时间用于数据分析”,这是一个本末倒置的事情。

需要说明的是:Excel及其Power Query、Excel数据分析与Python数据分析、Python中的openpyxl库与xlwings库等,这些都是解决问题的工具。对问题的解决而言,工具的利用可以在最优解与次佳方案之间抉择,也可以让二者进行协同作业,从而使各工具的效能最大化。

3.1.3 Excel数据的读取流程

在Pandas中,可以通过pd.read_csv()、pd.read_excel()、pd.ExcelFile.parse()、pd.read_json()、pd.read_sql()等众多方式读取来自不同数据源的数据。在本章节将详细列举使用pd.read_excel()或pd.ExcelFile.parse()将Excel文件读取到DataFrame。这两个函数及解析方法参数众多,对于初学者不易理解与记忆。考虑到Pandas读取与转换Excel数据时与Excel中的Power Query读取与转换有一些类似的地方,现采用对比学习的方式,对相关读取流程做一些简要的整理,顺带了解Power Query这门常用且易用的数据分析工具:

(1)在微软Excel的Power Query中,用户通常会通过“数据”→“获取数据”→“自文件”→“从Excel工作簿”获取来自指定路径的工作簿;或者可以通过“数据”→“来自表格/区域”,获取来自当前工作簿中的数据。相当于Pandas中pd.read_excel()或pd.ExcelFile()函数中的io参数,如图3-2所示。

图3-2 利用Power Query获取Excel工作簿(1)

如果使用的是微软Power BI中的Power Query,则在启运Power BI后,通常会通过选择功能区的“主页”→“Excel工作簿”或单击报表画布区的“从Excel导入数据”获取Excel工作簿,如图3-3所示。

图3-3 利用Power Query获取Excel工作簿(2)

(2)该步骤及后续的操作步骤,在Excel或Power BI的Power Query中大体相似,不过Power BI中的Power Query功能更为强大。当Power Query读取的工作簿中包含多个工作表时,可以在导航器窗格中选择全部、部分或某工作表;该步骤相当于Pandas中sheet_name参数的应用,默认为只选择第1个工作表,如果为None,则为选择全部工作表,当然也可以通过列表的方式,选择部分工作表,例如,sheet_name=[0,1,'合肥']。Power Query导航器窗格中工作表的选择如图3-4所示。

图3-4 在Power Query导航器窗格中选择工作表

(3)在Power Query中,在默认情况下会将数据的首行提升为标题。这相当于Pandas中header参数的应用。

(4)在Power Query中,在将首行提升为标题后,系统会默认识别并自动更改数据的类型。这相当于Pandas中dtype及converters参数的应用。

(5)在Power Query中,如果只想导入需要的列,则可以通过“选择列”或“删除其他列”的方式导入。这相当于Pandas中usecols参数的应用。

(6)在Power Query中,当前面几行为空行时,可通过“主页”→“删除行”→“删除前面几行”,然后将首行提升为标题。这相当于Pandas中skiprows参数的应用,而nrows参数则类似于“主页”→“保留行”→“保留最前面几行”的应用;skipfooter参数则类似于“主页”→“保留行”→“保留最后几行”的应用。其他不再一一举例。Power Query编辑器中的图形化操作如图3-5所示。

图3-5 在Power Query中删除行

3.2 pd.read_excel()函数

pd.read_excel()函数是Pandas中将Excel读取到DataFrame的一个函数。该函数共有20多个参数,除第1个参数io外,其他参数均为可选参数。该函数适用于各类复杂应用场景,功能十分强大。它可以读取来自本地端及URL中的xls、xlsx、xlsm、xlsb等格式的文件,支持读取工作簿中单个或多个工作表;如果为多个工作表,则各工作表的名称必须存放于一个列表中,其语法如下:

    pd.read_excel(io,sheet_name = 0,*,header = 0,names = None,index_col = None,usecols =
    None,squeeze = None,dtype = None,engine = None,converters = None,true_values = None,false_
    values = None,skiprows = None,nrows = None,na_values = None,keep_default_na = True,na_
    filter = True,verbose = False,parse_dates = False,date_parser = None,thousands = None,
    decimal = '.',comment = None,skipfooter = 0,convert_float = None,mangle_dupe_cols = True,
    storage_options = None)[source]

3.2.1 路径参数

pd.read_excel()函数的第1个参数(io)可以为有效的字符串路径、Excel文件、带Excel工作簿的URL等。第1个参数io可以是当前路径、相对路径或绝对路径。

(1)采用当前路径表示,代码如下:

    pd.read_excel('华东.xlsx')

(2)采用相对路径表示,代码如下:

    pd.read_excel(r'.\华东.xlsx')

(3)采用绝对路径表示,代码如下:

    pd.read_excel(r'D:\数据源\B文件\华东.xlsx')

在以上代码中r为转义字符。本书后续章节主要采用文件的绝对路径进行讲解。另外,本书95%以上案例演示所用的数据源为工作簿(订单表.xlsx)及其子集(华东.xlsx),子集的拆分过程详见第10章(ch10d041.py)。本章第4节和第5节示例工作簿(上海.xlsx)是工作簿(华东.xlsx)的子集。

3.2.2 引擎参数

Pandas读取Excel工作簿时所支持的引擎为xlrd(.xls格式)、openpyxl(.xlsx等最新版本格式)、odf(odf、ods、odt格式)、pyxlsb(二进制Excel文件)。Pandas在读取xls时调用xlrd,在读取xlsx时调用xlrd或openpyxl。在写入Excel时调用xlwt、openpyxl或xlswriter。

在Anaconda安装环境中,系统已自动安装并配置好了以上Pandas依赖项。在默认情况下,该参数的值将设置为None,系统会自动检测并使用适配的engine。

3.3 工作表名称

pd.read_excel()函数的sheet_name参数的默认值为0。①在Pandas中,sheet的索引值是从0开始的,0代表的是工作簿中的第1个工作表(sheet),1代表的是第2个sheet。②sheet_name的参数可以用0、1等整数表示,也可以用“Sheet1、Sheet2”等字符串表示。③当sheet_name导入的是一个列表时,可以存在整数与字符串混用的情形,例如[0,1,'Sheet3'],返回的值是一个DataFrame字典。④当sheet_name=None时,获取的是工作簿中所有工作表的内容。

3.3.1 选择工作簿中的所有工作表

当sheet_name=None时,获取的是工作簿中所有的工作表,代码如下:

    #ch03d001.py
    import pandas as pd
    df = pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name = None)
    type(df)

返回的值为dict。采用列表推导式获取工作簿中所有的工作表名,代码如下:

    [i for i in pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name = None)]

返回的值如下:

    ['上海','南京','厦门','合肥','宁波','常州','无锡','杭州','苏州']

获取字典中的指定DataFrame,代码如下:

    df1 = pd.read_excel(
          r'D:\数据源\B文件\华东.xlsx',
          sheet_name = None
      )
    df1['上海']

返回的值如图3-6所示。

图3-6 获取工作簿中的工作表(上海)

3.3.2 选择工作簿中指定的工作表

pd.read_excel()函数的sheet_name参数可为str(字符串)、int(整数)、list(列表)。以sheet_name=[0,1]为例,获取工作簿中的第1个和第2个工作表,代码如下:

    df2 = pd.read_excel(
          r'D:\数据源\B文件\华东.xlsx',
          sheet_name = [0,1]
      )
    df2

第2个参数的sheet_name=[0,1]可简写成[0,1],读取的是第1个和第2个sheet,生成的是1个字典,键为0和1,值为两个DataFrame。返回的值如下:

    {0: 片区 订单来源 接单时间 订单编号 包装方式    产品 订单数 入库日期 入库数
    0 华东  上海2020-06-30 DD003   箱装 苹果醋   2 2020-07-03 00:03:21  2
    1 华东  上海2020-08-02 DD005   桶装   油漆  2 2020-08-02 09:07:18  2
    2 华东  上海2021-10-01 DD028   散装 钢化膜   4 2021-10-03 07:24:21  4,
    1: 片区 订单来源 接单时间 订单编号 包装方式   产品 订单数 入库日期 入库数
    0 华东 南京2020-10-02 DD011   桶装   油漆  9 2020-10-02 06:22:05  9
    1 华东 南京2021-10-11 DD029  袋 老陈醋  9 2021-10-30 08:09:16  9}

此时可通过df2[0]获取第1个DataFrame,可通过df2[1]获取第2个DataFrame sheet_name列表中的值可以是整型数值和字符串的混合,代码如下:

    df3 = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       sheet_name = [0,1,'无锡']
    )
    df3

返回的是1个字典,键为“0、1、无锡”,内容如下:

    {0: 片区 订单来源 接单时间 订单编号 包装方式     产品 订单数 入库日期 入库数
    0 华东   上海 2020-06-30 DD003  箱装 苹果醋   2 2020-07-03 00:03:21  2
    1 华东   上海 2020-08-02 DD005  桶装   油漆 2 2020-08-02 09:07:18  2
    2 华东   上海 2021-10-01 DD028  散装 钢化膜  4 2021-10-03 07:24:21  4,
    1:   片区 订单来源 接单时间 订单编号 包装方式    产品 订单数 入库日期 入库数
    0 华东   南京 2020-10-02 DD011  桶装  油漆   9 2020-10-02 06:22:05  9
    1 华东   南京 2021-10-11 DD029  袋 老陈醋   9 2021-10-30 08:09:16  9,
    '无锡': 片区 订单来源 接单时间  订单编号 包装方式   产品 订单数 入库日期 入库数
    0 华东   无锡 2021-07-27 DD018  膜 异形件   2 2021-07-28 06:42:49  2}

3.3.3 选择工作簿中具体的某一工作表

当工作簿中存在多个工作表时,如果未对工作表的位置或工作表名称进行指定,则默认导入的是第1个工作表。也可以通过具体的指定位置或表名获取具体的某一工作表,代码如下:

    df4 = pd.read_excel(r'D:\数据源\B文件\华东.xlsx',7)
    df4

在华东.xlsx工作簿中,第8个工作表的名称为“杭州”,所以也可以采用以下代码:

    df5 = pd.read_excel(r'D:\数据源\B文件\华东.xlsx','杭州')
    df5

df4及df5返回的值如图3-7所示。

图3-7 获取工作簿中的工作表(杭州)

3.4 标题

pd.read_excel()函数的标题参数为header,可设置为int、list of int,默认值为0。

3.4.1 提升为标题

pd.read_excel()函数的header参数可为str(字符串)、int(整数)、list(列表)。在默认情况下,将第1行提升为标题,即header=0,代码如下:

    #ch03d002.py
    import pandas as pd
    df1 = pd.read_excel(r'D:\数据源\B文件\上海.xlsx')
    df1

返回的值如图3-8所示。

图3-8 将第1行提升为标题

3.4.2 不需要标题

如果数据源中不存在标题,则可以将header设置为None(不需要标题),代码如下:

    df2 = pd.read_excel(r'D:\数据源\B文件\上海.xlsx',header=
    None)
    df2

返回的值如图3-9所示。

图3-9 不需要标题

对于确实不存在列名的数据可以通过name参数命名列。names参数可为array-like,默认值为None。利用names构建列名,代码如下:

    df3 = pd.read_excel(
       r'D:\数据源\B文件\上海.xlsx',
       header = None,
       names = ['片区','订单来源','接单时间','订单编号']
    )
    df3

将以上names的列表改成元组也是允许的,代码如下:

    tpl = ('片区','订单来源','接单时间','订单编号')
    df4 = pd.read_excel(
       r'D:\数据源\B文件\上海.xlsx',
       header = None,
       names = tpl
    )
    df4

在日常操作时,为了防止数值被提升为标题,在Pandas数据导入的过程中,经常会先设置header=None,然后通过names来指定标题/列名。以上代码返回的值如图3-10所示。

当数据的标题位于第3行时,可以将第3行(header=2)指定为标题,代码如下:

    df5 = pd.read_excel(
       r'D:\数据源\B文件\上海.xlsx',
       sheet_name = 1,
       header = 2)
    df5

图3-10 利用names参数构建列名

返回的值如图3-11所示。

图3-11 将指定的行提升为标题

3.4.3 多层索引

当Excel中存在合并单元格及多级标题情况时,在Pandas中可以利用list of int来构建多层索引,代码如下:

    df6 = pd.read_excel(
       r'D:\数据源\B文件\上海.xlsx',
       sheet_name = 2,
       header = [0,1])
    df6

返回的值如图3-12所示。

图3-12 构建多层索引

3.5 处理行

pd.read_excel()函数与行相关的参数有skiprows、nrows、skipfooter。

3.5.1 跳过行

skiprows参数可为int、list-like或callable。当参数为callable时,参数允许为函数(含自定义函数及匿名函数)。以skiprows=list-like为例,跳过前两行,代码如下:

    #ch03d003.py
    import pandas as pd
    df1 = pd.read_excel(
           r'D:\数据源\B文件\上海.xlsx',
           sheet_name = 1,
           skiprows = [0,1]
    )
    df1

返回的值如图3-13所示。

图3-13 跳过行(1)

如果需要跳过的行位于标题前及数据中,则也是允许的,代码如下:

    df2 = pd.read_excel(
           r'D:\数据源\B文件\上海.xlsx',
           sheet_name = 3,
           skiprows = [0,1,4,6]
    )
    df2

返回的值如图3-14所示。

图3-14 跳过行(2)

skiprows参数用于跳过前面的行。如果需要跳过后面的行,则需要用skipfooter参数,该参数的值为int数据类型,代码如下:

    df3 = pd.read_excel(
           r'D:\数据源\B文件\上海.xlsx',
           sheet_name = 2,
           header = [0,1],
           skipfooter = 1
    )
    df3

返回的值如图3-15所示。

图3-15 跳过行(3)

3.5.2 保留行

如果需要保留数据中的最前面几行,则可以用nrows参数,nrows的值为int数据类型。以保留最前面两行数据为例,代码如下:

    df4 = pd.read_excel(
           r'D:\数据源\B文件\上海.xlsx',
           sheet_name = 3,
           skiprows = [0,1,4,6],
           nrows = 2
    )
    df4

返回的值如图3-16所示。

Pandas的行筛选功能相当强大,这些会放在后续章节重点讲解。

图3-16 保留行

3.6 选择列

pd.read_excel()函数的选择列参数为usecols,可以采用位置索引(整型数字)或列名索引(文本字符串)方式进行列的选择。可设置为str、list-like或callable,默认值为None。当未填写usecols或指定usecols=None时,选择所有列。

3.6.1 指定索引列

index_col的默认值为None,可选str、int、list of int,生成索引。

应用举例,index_col的值为字符串,代码如下:

    #ch03d004.py
    import pandas as pd
    df1 = pd.read_excel(
        r'D:\数据源\B文件\订单表.xlsx',
        index_col = '片区'
    )
    df1.loc['西南']

或index_col的值为int,代码如下:

    df2 = pd.read_excel(
        r'D:\数据源\B文件\订单表.xlsx',
        index_col = 0
    )
    df2.loc['西南']

返回的值如图3-17所示。

图3-17 指定索引列(1)

index_col的值为list of int,代码如下:

    df3 = pd.read_excel(
        r'D:\数据源\B文件\订单表.xlsx',
        index_col = [0,1]
    )
    df3.head()

df.head()函数用于获取前5条数据,head()函数的括号内可填写需获取的具体条数;df.tail()函数用于获取后5条数据。返回的值如图3-18所示。

图3-18 指定索引列(2)

3.6.2 选择列

采用字符串方式,字符串对应的是Excel工作表中的列字母与列范围(例如'A: D'或'A, :D')。当默认值为None时解析工作表中所有的列,代码如下:

    #ch03d005.py
    import pandas as pd
    df1 = pd.read_excel(
        r'D:\数据源\B文件\华东.xlsx',
        usecols = 'A:D', #采用字符串方式
    )
    df1

或采用list-like方式,列表内容全为文本,代码如下:

    df2 = pd.read_excel(
        r'D:\数据源\B文件\华东.xlsx',
        usecols = ['片区','订单来源','接单时间','订单编号',]  #可读性最好
    )
    df2

或者,列表内容全为整数。在列表中,最后一个逗号有无不影响输出,代码如下:

    df3 = pd.read_excel(
        r'D:\数据源\B文件\华东.xlsx',
        usecols = [0,1,2,3]               #不能采用[:4]或[0:4]切片方式
    )
    df3

或事先预设的列表的字段顺序,代码如下:

    ucl = ['片区','订单来源','接单时间','订单编号',]
    df4 = pd.read_excel(
        r'D:\数据源\B文件\华东.xlsx',
        usecols = ucl
    )
    df4

或事先预设元组,代码如下:

    tpl = ('片区','订单来源','接单时间','订单编号',)
    df5 = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       usecols = tpl
    )
    df5

以上代码返回的值如图3-19所示。

图3-19 选择列(1)

当usecols参数为list-like时,不可以采用整数与字符串混合的方式(例如[0,1,2,'订单编号'])。当采用cols=[0,1,2,'订单编号']时,会出现的错误提示如下:

    ValueError:'usecols   'must   either  be  list-like  of  all
    strings,all unicode,all integers or a callable.

当采用字符串方式时,字符串可以来自不连续的区域,代码如下:

    df6 = pd.read_excel(
           r'D:\数据源\B文件\华东.xlsx',
           usecols = 'A,C:D',
    )
    df6

返回的值如图3-20所示。

图3-20 选择列(2)

当usecols采用的是数值列表时,字符串可以来自不连续的区域,代码如下:

    df7 = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       usecols = range(1,8,2),
    )
    df7

返回的值如图3-21所示。

图3-21 选择列(3)

当usecols采用的是匿名函数时,代码如下:

    df8 = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       usecols = lambda x:~x.find('单')
    )
    df8

返回的值如图3-22所示。

图3-22 选择列(4)

3.7 数据类型

在Pandas中,数据类型有number、float、int、object、datetime、timedelta、object、category等数据类型,也可以在代码中使用Python或NumPy等效的数据类型,或者对float、int数据类型进行8位、16位、32位、64位、128位细分指定。

3.7.1 查看数据类型

利用Pandas的dtypes属性对DataFrame进行属性查看,代码如下:

    #ch03d006.py
    import pandas as pd
    df = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       usecols = [2,3,5,6,8]
    )
    df.dtypes

返回的值如下:

    接单时间  datetime64[ns]
    订单编号      object
    产品       object
    订单数       int64
    入库数       int64
    dtype:object

在Pandas中,int与float数据类型默认为64位。

pd.read_excel()函数可采用字典映射方式对读取的数据类型进行指定或调整,代码如下:

    type = {"接单时间":'datetime64[ns]',"订单数":'int8','入库数':'float'}
    df1 = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       usecols = [2,3,5,6,8],
       dtype = type
    )
    df1

返回的值如图3-23所示。

图3-23 调整数据类型后的值

利用dtypes属性,查看DataFrame的数据类型,代码如下:

    df1.dtypes

返回的值如下:

    接单时间  datetime64[ns]
    订单编号      object
    产品       object
    订单数         int8
    入库数      float64
    dtype:object

入库数的数据类型由int64调整为float64。

如果对DataFrame中Series的数据类型进行调整,则可以采用astype()方法,代码如下:

    df1['入库数'] = df1['入库数'].astype(int)
    df1.dtypes

返回的值如下:

    接单时间  datetime64[ns]
    订单编号      object
    产品       object
    订单数        int8
    入库数       int32
    dtype:object

入库数的数据类型由float64调整为int32。

3.7.2 转换数据类型

pd.read_excel()函数converters参数的默认值为None。采用的是字典映射方式对数据类型进行指定或对数据进行修改,字典映射方式为dict{str, Callable}或dict{int, Callable},代码如下:

或者

返回的值如图3-24所示。

图3-24 转换数据类型

利用DataFrame的dtypes属性,查看数据类型,代码如下:

    df1.dtypes

返回的值如下:

    接单时间    object
    订单编号    object
    产品     object
    订单数     int64
    入库数    float64
    dtype:object

接单时间的数据类型由datetime被调整为object;入库数的数据类型由int被调整为float。

3.7.3 解析日期时间列

parse_dates参数用于解析日期时间列,默认值为False。可采用parse_dates=True、parse_dates=[列名]、parse_dates=[列位置]或parse_dates=[[]]的索引方式。①parse_dates=True用于尝试将索引列的文本日期时间解析为日期时间;②当索引列不是日期时间时,可采用parse_dates=['接单时间','入库日期']或parse_dates=[2,7]进行解析;③当年、月、日、时、分、秒等数值型数据分别存放于各列时,可采用类似parse_dates=[['年','月','日']]的二维列表形式将其合并成一列,各列的列名或位置索引均存放于内层列表中。如果需要规范拼接后的列名,则可以采用字典形式,类似于parse_dates={'日期':['年','月','日']},拼接后的列名为日期。

date_parse参数用于指定解析日期时间的格式。在一般情况下,当使用了parse_dates参数时,会自动调用date_parse参数对各类文本型日期进行格式解析。

3.8 对Excel的读取与解析

ExcelFile()是Pandas中对Excel表格文件进行读取操作的类,用于将Excel文件读取到Pandas DataFrame中,特别适用于含有多个Sheet的Excel,支持Excel早期版本的xls格式。ExcelFile()与WriteExcel()经常搭配使用,实现对多个Sheet的读写,非常方便与快捷,语法如下:

    ExcelFile(io,storage_options = storage_options,engine = engine).parse(sheet_name = 0,header = 0,
    names = None,index_col = None,usecols = None,squeeze = False,converters = None,true_values =
    None,false_values = None,skiprows = None,nrows = None,na_values = None,parse_dates = False,
    date_parser = None,thousands = None,comment = None,skipfooter = 0,convert_float = True,
    mangle_dupe_cols = True, * * kwds)

当sheet_name=None时,获取的是工作簿中所有的工作表,代码如下:

    #ch03d008.py
    import pandas as pd
    df = pd.ExcelFile(r'D:\数据源\B文件\华东.xlsx').parse()
    df

返回的值如图3-25所示。

图3-25 获取工作簿中的第1个工作表

采用parse()方法解析路径文件。以sheet_name=None为例,代码如下:

    df1 = pd.ExcelFile(r'D:\数据源\B文件\华东.xlsx').parse(sheet_name=None)
    dict(df1)

返回的值为dict。采用列表推导式了解工作簿中所有的工作表名,代码如下:

    [i for i in df1]

返回的值如下:

    ['上海','南京','厦门','合肥','宁波','常州','无锡','杭州','苏州']

从中不难发现,以下两种写法返回的值相同。

    #方法一
    pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name = None)
    #方法二
    pd.ExcelFile(r'D:\数据源\B文件\华东.xlsx').parse(sheet_name = None)

因此,当pd.read_excel()函数中与pd.ExcelFile.parse()函数中存在相同的参数,其参数的用法基本是相同的。

3.9 存储数据

在日常使用过程中,常常会将pd.read_excel()函数与pd.to_excel()函数搭配使用,还会将pd.ExcelFile()函数与pd.ExcelWriter()函数搭配使用。df.to_excel()函数仅支持将单个Sheet表格写入Excel,而pd.ExcelWriter()函数支持将多个Sheet表格写入Excel。

3.9.1 to_excel

在Pandas中,可以通过df.to_csv()、df.to_excel()、pd.ExcelWriter()、df.to_json()、df.to_sql()等众多数据的写入方式。df.to_excel()方法用于将DataFrame写入Excel工作簿。如果只需将单个DataFrame写入Excel工作簿文件,则需要指定Excel文件名;如果需要在Excel中写入多个工作表,则需要先创建pd.ExcelWriter对象,并在文件中指定要写入的工作表,语法如下:

    DataFrame.to_excel(excel_writer,sheet_name = 'Sheet1',na_rep = '',float_format = None,
    columns = None,header = True,index = True,index_label = None,startrow = 0,startcol = 0,
    engine = None,merge_cells = True,encoding = None,inf_rep = 'inf',verbose = True,freeze_panes =
    None,storage_options = None)

应用举例,指定列、空值显示的值、行与列的偏移位置等,代码如下:

    #ch03d009.py
    import pandas as pd
    #获取数据
    df1 = pd.read_excel(r'D:\数据源\B文件\华东.xlsx')
     
    #存储数据
    df1.to_excel(r'D:\数据源\第3章\ch0309\(华东)上海.xlsx',
             na_rep = '--',
             columns = ['订单编号','订单来源','入库数','订单数'],
             startrow = 3,
             startcol = 1)

文件的存储路径为D:\数据源\第3章\写入\(华东)上海.xlsx。返回的值如图3-26所示。

图3-26 将DataFrame写入Excel工作簿

3.9.2 ExcelWriter

pandas.ExcelWriter()函数用于在同一工作簿中保存多个工作表。应用举例,将华东.xlsx工作簿中的苏州、杭州工作表的数据存储到苏杭.xlsx文件中对应的工作表,代码如下:

    df2 = pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name='苏州')
    df3 = pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name='杭州')
    with pd.ExcelWriter(r'D:\数据源\第3章\ch0309\苏杭.xlsx')as wt:
       df2.to_excel(wt,sheet_name = '苏州'),
       df3.to_excel(wt,sheet_name = '杭州')

文件被保存在苏杭.xlsx工作簿中,该工作簿中有苏州、杭州两个工作表。在以上代码中with语句用于将文件的写操作包含在一个代码内,当执行完成后就会自动关闭文件对象,无须再调用close()方法。

应用举例,批量读取工作簿(华东.xlsx)中所有工作表的数据,并批量提取列数据,代码如下:

    dfs = pd.read_excel(
       r'D:\数据源\B文件\华东.xlsx',
       sheet_name = None,
       usecols = ['片区','订单来源','接单时间','订单编号'],
      )
     
    with pd.ExcelWriter(r'D:\数据源\第3章\ch0309\华东A.xlsx')as wt:
       for s,df in dfs.items():
          df.to_excel(wt,sheet_name = s,index = False)

或者可采用以下代码:

    dfs = pd.read_excel(r'D:\数据源\B文件\华东.xlsx',sheet_name=None)
     
    with pd.ExcelWriter(r'D:\数据源\第3章\ch0309\华东B.xlsx')as wt:
       for s,df in dfs.items():
          cols = df[['片区','订单来源','接单时间','订单编号']]
          cols.to_excel(wt,sheet_name = s,index = False)

运行以上代码,返回的值如图3-27所示。

图3-27 将DataFrame批量写入Excel工作簿 eALY9sxCfpJ5oJn5yWc3CTXfUoyvSMa5Z6G7VwVut5w/2saskj/PBSR0H1kOV1Ko

点击中间区域
呼出菜单
上一章
目录
下一章
×