安装xlrd模块和xlwt模块
读取Excel文件了内容需要额外的模块—— \(xlrd\),在官网上可以找到下载:https://pypi.python.org/pypi/xlrd#downloads。
下载完毕后解压到当前文件夹,点击进入,可以找到一个 setup.py 文件,在当前目录下打开 cmd 窗口,运行“python setup.py install”(需要提前将python添加环境变量),成功后,去Python Shell中运行“import xlrd”试一下,应该没什么问题。解压得到的xlrd模块文件都可以删除。
另外,如果你使用PyCharm的话,那就更好办了,点击"File" -> "Settings" -> "Project interpreter" -> "+" -> 查找"xlrd" -> "install package",OK了。
同样的,写入Excel文件需要的是 \(xlwt\) 模块,方法类似,就不多解释了。下载地址:https://pypi.python.org/pypi/xlwt。
读取Excel文件
不得不说,这个模块简单易懂,容易上手,几分钟就会了。直接看代码如何使用吧!
class myExcel(object):
def __init__(self, excelPath):
self.path = excelPath
def readExcel(self):
excel = xlrd.open_workbook(self.path)
# get sheet names
print(excel.sheet_names())
# get sheet content(By index or By name)
sheet = excel.sheet_by_index(0)
sheet1 = excel.sheet_by_name("Sheet1")
sheet2 = excel.sheets()[0]
# get sheet info
print(sheet.name, sheet.nrows, sheet.ncols)
# get row/col content
row = sheet.row_values(0)
col = sheet.col_values(0)
print(row, col)
# get cell value
cell = sheet.cell(0, 0)
print(cell.value)
print(sheet.cell_value(0, 0))
print(sheet.row(0)[0].value)
# get cell ctype:0empty 1string 2number 3date 4boolean 5error
print(sheet.cell(0, 0).ctype)
return excel
def printExcel(self):
excel = self.readExcel()
sheet = excel.sheets()[0]
for i in range(0, sheet.nrows):
row = sheet.row_values(i)
for item in row:
print(item, '|')
写入Excel文件
这个也还行,单纯的加入数据还是挺简单的。
发现了一个问题,这个xlwt模块只能保存为".xls"表格,有点绝望。
稍微查了一下,果然存在可以保存为".xlsx"表格的模块,叫做 \(openpyxl\),请参考:https://www.cnblogs.com/anpengapple/p/6399304.html?utm_source=itdadao&utm_medium=referral
def writeOneRow(self, sheet, nrow, rowList):
# 格式:{单元格:背景黄色;字体:红色粗体}
style = xlwt.easyxf('pattern: pattern solid, fore_colour yellow;font: bold 0, color red;')
ncol = 0
for row in rowList:
sheet.write(nrow, ncol, row, style)
ncol += 1
def writeExcel(self, excelName):
# create a Excel and add sheet
excel = xlwt.Workbook()
sheet = excel.add_sheet('sheet1', cell_overwrite_ok=True)
headList = ['Stu_id', 'name', 'sex']
data = [['1', 'stu1', '男'],
['2', 'stu2', '男'],
['3', 'stu3', '女'],]
self.writeOneRow(sheet, 0, headList)
for nrow in range(0, 3):
self.writeOneRow(sheet, nrow, data[nrow])
excel.save(excelName)
Excel转换成Markdown格式
其实这是我做这个小练习的主要原因,由于随笔都是用Markdowm格式写的,有一篇随笔是LeetCode所有题目的列表:
LeetCode All in One题解汇总(持续更新中...)。我才不想一条一条加呢!稍微转换一下,添加点格式符号就OK了。
def toMarkdown(self, mdName):
excel = self.readExcel()
sheet = excel.sheets()[0]
file = open(mdName, 'w', encoding='utf-8')
rowStr = "| %d | [%s]() | %.3f | '%s' |\n"
try:
file.write('| 编号 | 题名 | 过题率 | 难度 |' + '\n')
file.write('| :------------: |:---------------: | :------------:| :-----:|' + '\n')
for i in range(0, sheet.nrows):
row = sheet.row_values(i)
data = rowStr % (int(row[0]), str(row[1]), float(row[2]), str(row[3]))
file.write(data)
print('成功写入文件,共有%d条记录...' % sheet.nrows)
except Exception as e:
print(e)
finally:
file.close()
完整代码:https://github.com/Pacsiy/learnPY/tree/master/LeetCodeAllProblem
本文版权归作者AlvinZH和博客园所有,欢迎转载和商用,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利.