一、概述
操作 excel 是程序员经常要遇到的场景。因为产品、运营的数据都是以这种格式存储。所以,当程序员拿到这些数据肯定要解析,甚至需要把结果输出成 excel 文件。
下面就介绍如果用 Python 方面的读、写 excel 文件。
二、openpyxl
A Python library to read/write Excel 2010 xlsx/xlsm files
借助 Python 的三方库 openpyxl
,让操作 excel 变得简单。
# coding=utf-8
from openpyxl import Workbook
wb = Workbook()
# 选择 sheet
ws = wb.active
# 设置值到某一个单元格(cells)
ws['A1'] = 42
# Python 的数据类型可以自动转换
import datetime
ws['A2'] = datetime.datetime.now()
# 存储文件
wb.save("sample.xlsx") # 默认保存到当前目录下。文件名称为 sample.xlsx
读数据
from openpyxl import load_workbook
wb = load_workbook('sample.xlsx') # 读取文件
sheet = wb.get_sheet_by_name("Sheet") # 根据 sheet 名称获取,返回 Worksheet 对象
columns = sheet['A'] # 选择一列
for fi_column in columns:
# 遍历这列的所有行
print fi_column.value # 每一个fi_column是 Cell 对象
写数据
from openpyxl import Workbook
wb = Workbook()
# 选择 sheet
ws = wb.create_sheet()
# result_list ->[[第一行数据],[第二行数据],...]
for fi_result in result_list:
ws.append(fi_result) # 每行的数据
# 存储文件
wb.save("test.xlsx")