在python自动化中,经常会遇到对数据文件的操作,比如添加多名员工,但是直接将员工数据写在python文件中,不但工作量大,要是以后再次遇到类似批量数据操作还会写在python文件中吗?

  应对这一问题,可以将数据写excel文件,针对excel 文件进行操作,完美解决。

  本文仅介绍python对excel的操作

  

安装xlrd 库

  xlrd库 官方地址:https://pypi.org/project/xlrd/

  pip install xlrd

    

  笔者在安装时使用了 pip3 install xlrd

  原因:笔者同时安装了python2 和 python3

  如果pip的话会默认将库安装到python2中,python3中不能直接调用。

  那么到底是使用pip 还是pip3进行安装呢?

    如果系统中只安装了Python2,那么就只能使用pip。
    如果系统中只安装了Python3,那么既可以使用pip也可以使用pip3,二者是等价的。
    如果系统中同时安装了Python2和Python3,则pip默认给Python2用,pip3指定给Python3用。

 

Xlrd 库简单的使用

  以如下excel文件为例进行操作

  文件名为demo,有两个sheet,名为工作表1和工作表2

  工作表1中有如下数据

    

简单的使用  

# coding=utf-8

import xlrd

# 打开文件
data = xlrd.open_workbook('file/demo.xlsx')

# 查看工作表
data.sheet_names()
print("sheets:" + str(data.sheet_names()))

# 通过文件名获得工作表,获取工作表1
table = data.sheet_by_name('工作表1')

# 打印data.sheet_names()可发现,返回的值为一个列表,通过对列表索引操作获得工作表1
# table = data.sheet_by_index(0)

# 获取行数和列数
# 行数:table.nrows
# 列数:table.ncols
print("总行数:" + str(table.nrows))
print("总列数:" + str(table.ncols))

# 获取整行的值 和整列的值,返回的结果为数组
# 整行值:table.row_values(start,end)
# 整列值:table.col_values(start,end)
# 参数 start 为从第几个开始打印,
# end为打印到那个位置结束,默认为none
print("整行值:" + str(table.row_values(0)))
print("整列值:" + str(table.col_values(1)))

# 获取某个单元格的值,例如获取B3单元格值
cel_B3 = table.cell(3,2).value
print("第三行第二列的值:" + cel_B3)

运行后结果

 

项目中使用

  获得所有的数据

 1 # coding=utf-8
 2 
 3 import xlrd
 4 
 5 def read_xlrd(excelFile):
 6     data = xlrd.open_workbook(excelFile)
 7     table = data.sheet_by_index(0)
 8 
 9     for rowNum in range(table.nrows):
10         rowVale = table.row_values(rowNum)
11         for colNum in range(table.ncols):
12             if rowNum > 0 and colNum == 0:
13                 print(int(rowVale[0]))
14             else:
15                 print(rowVale[colNum])
16         print("---------------")
17 
18     # if判断是将 id 进行格式化
19     # print("未格式化Id的数据:")
20     # print(table.cell(1, 0))
21     # 结果:number:1001.0
22 
23 
24 if __name__ == '__main__':
25     excelFile = 'file/demo.xlsx'
26     read_xlrd(excelFile=excelFile)

  结果

  

  如果在项目中使用则可将内容方法稍为做修改,获得所有的数据后,将每一行数据作为数组进行返回  

 1 # coding=utf-8
 2 
 3 import xlrd
 4 
 5 def read_xlrd(excelFile):
 6     data = xlrd.open_workbook(excelFile)
 7     table = data.sheet_by_index(0)
 8     dataFile = []
 9 
10     for rowNum in range(table.nrows):
11         # if 去掉表头
12         if rowNum > 0:
13             dataFile.append(table.row_values(rowNum))
14 
15     return dataFile
16 
17 
18 if __name__ == '__main__':
19     excelFile = 'file/demo.xlsx'
20     print(read_xlrd(excelFile=excelFile))

  结果

  

版权声明:本文为tynam原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/tynam/p/11204895.html