(1) 如果excel文件是xls,2003版的,使用xlrd和xlwt库来对xls文件进行操作
(2) 如果excel文件是xlsx,2007以上版的,使用openpyxl库来对xlsx文件进行操作
Tips:xlrd、xlwt和openpyxl非python自带库,需要进行安装,可以查看我之前的文章。
如果你在学习Python的过程当中有遇见任何问题,可以加入我的python交流学习qq群:683380553,多多交流问题,互帮互助,群里有不错的学习教程和开发工具。学习python有任何问题(学习方法,学习效率,如何就业),可以随时来咨询我,如果你准备学习大数据,也欢迎加入大数据学习交流qq群683380553,每天与大家分享学习资源哦。
我们使用Python做数据挖掘和分析时候,当数据行超过一定数量,xls文件是存不下的。显然无法满足我们的大量数据存储需求,这个时候需要改用xlsx。
那具体xls和xlsx最大分别可以存多少行呢?
(1) 对于2003版本的xls文件,最大行数是65536行
(2) 对于2007版本的xlsx文件,最大行数是1048576行
闲话不多聊,直接上代码干货
#!/usr/bin/env python3
–– coding: utf-8 ––
读写2007 excel
import pprint
import openpyxl
import openpyxl.styles
from openpyxl.styles import Font,colors
读取Excel文件
def readExcel(filename):
workbook = openpyxl.load_workbook(filename)
worksheet = workbook[‘用户’]
list = []
for row in worksheet.rows: # 1行1行读
lineData = [] # 行数据
col = 1
for cell in row:
lineData.append(cell.value) # 1列1列读
col = col + 1
list.append(lineData)
return list
操作数据
def operateData(filename):
list = readExcel(filename)
# 去除第一行,第一行是表头
list.pop(0)
pprint.pprint(list)
pprint.pprint(“先按性别排序,再按身高排序”)
# 要对身高进行排序,但是男女有别
# 先按性别排序,再按身高排序
list.sort(key=lambda ele: (ele[3], ele[4]))
pprint.pprint(list)
获取Excel标签列表
比如第5行标签列表,总共4列 [‘A5’, ‘B5’, ‘C5’, ‘D5’]
def getTagList(index, colNum):
tagList = []
for i in range(0, colNum):
# A的ascii码值65
tag = chr(65 + i) + str(index)
tagList.append(tag)
return tagList
写入Excel文件
def writeExcel(outputFilePath, list):
book = openpyxl.Workbook()
sheet = book.create_sheet(“用户”, 0)
sheet.title = “用户” # sheet名称
rowNum = len(list)
try:
# 1行1行读取
for i in range(1, rowNum + 1): # 下标从1开始
datalist = list[i -1] # 读取1行
col = 1
# 1列1列写入
for data in datalist:
sheet.cell(i, col, data) #写入内容
col += 1
# 获取标签列表
tagList = getTagList(i, len(datalist))
font = Font(‘微软雅黑’, size = 14, color = ‘333333’)
# 设置单元格字体、字号、颜色
for tag in tagList:
sheet[tag].font = font # 设置字体
# 保存文件
book.save(outputFilePath)
except Exception as e: # 捕获异常
print(e)
主函数
if name == “main“:
print(“读取xlsx格式的数据”)
userList = readExcel(‘user.xlsx’)
print(“写入xlsx文件”)
writeExcel(“user2.xlsx”, userList)
print(“操作数据”)
operateData(‘user.xlsx’)
image