Python 操作 excel

python基础

浏览数:299

2019-10-4

一、概述

操作 excel 是程序员经常要遇到的场景。因为产品、运营的数据都是以这种格式存储。所以,当程序员拿到这些数据肯定要解析,甚至需要把结果输出成 excel 文件。

下面就介绍如果用 Python 方面的读、写 excel 文件。

二、openpyxl

A Python library to read/write Excel 2010 xlsx/xlsm files

借助 Python 的三方库 openpyxl ,让操作 excel 变得简单。

  1. 安装:pip install openpyxl
  2. 文档:官方文档
  3. 示例代码:
# 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")

更多 API

  • Worksheet.columns():获取 sheet 所有列
  • Worksheet.iter_cols():通过列截断
  • Worksheet.rows():获取 sheet 所有行
  • Worksheet.iter_rows():通过行截断
  • Worksheet.cell():操作单元格
  • Workbook.save():存储文件
  • workbook.Workbook.create_sheet():创建新的 sheet
  • Workbook.sheetnames():获取 sheet 名称

作者:削微寒