Python处理Excel文件(csv, xls, xlsx)

Excel文件格式主要有csv,xlsx和xlsx,对于不同的格式,我们使用不同的包来进行处理。

使用csv包处理csv文件

读取csv文件

1
2
3
4
5
6
import csv

with open('./data.csv', 'r', newline='', encoding='utf-8-sig') as csvfile:
spamreader = csv.reader(csvfile)
for row in spamreader:
print(row)

其中,encoding='utf-8-sig'是为了编码正常可以正确显示中文,spamreader中的每一个row为list格式,可以循环取出每个单元格的值。

写入csv文件

1
2
3
4
5
6
import csv

with open('./data.csv', 'w', newline='', encoding='utf-8-sig') as csvfile:
spamwriter = csv.writer(csvfile)
spamwriter.writerow(['HELLO', 'WORLD'])
spamwriter.writerows([('SuiXin', 'Blog'), ('https://suixinblog.cn', '2019', 'CSV', '文件')])

结果:

写入csv文件

:使用delimiter参数来指定数据的分隔符,默认为,

使用pandas包读写csv数据文件

如果csv文件是数据类的,那么使用pandas包读写数据会更方便。

读取csv数据存入DataFrame对象中

1
2
3
4
import pandas as pd

df = pd.read_csv("hou_all.csv", header=None, names=["CRIM", "ZN", "INDUS", "CHAS", "NOX", "RM", "AGE", "DIS", "RAD", "TAX", "PTRATIO", "B", "LSTAT", "MEDV"])
print(df)

结果:

参数:

  • sep=',':str,分隔符;
  • header=0:int,指定行数用作列名。如果文件不包含列名可以设置为None
  • names:array-like,指定列名。相当于对没有列名的DataFrame设置df.columns
  • squeeze=False:bool,如果文件值包含一列,返回一个Series;
  • prefix:str,在没有列名时,给列添加前缀。如:prefix='V'则列名为V0V1V2……
  • skiprows=None:list-like or int,需要忽略的行数;
  • nrows=None:int,需要读取的行数;
  • skip_blank_lines=True:bool,是否跳过空行。

将数据写入csv文件

1
2
3
4
5
6
import pandas as pd

d = {'NAME': ['TOM', 'SuiXin'], 'AGE': [12, 24]}
df = pd.DataFrame(d)
print(df)
df.to_csv('data1.csv', index=False)

结果:

pandas写入csv数据

参数:

  • sep=',':str,分隔符;
  • na_rep='':str,缺失值表示;
  • float_format=None:str,浮点数格式。如float_format='%.2f'
  • columns:sequence,需要写入的列;
  • header=True:bool,list of str,写入列名;
  • index=True:bool,写入行名。

使用openpyxl包处理xlsx文件

Excel文件有三层对象:工作薄、工作表和三元格,分别对应openpyxl包中的workbook、sheet和cell。

workbook操作

1
2
3
4
5
6
7
8
# 读取xlsx文件
wb = openpyxl.load_workbook('test.xlsx')
# 以只读模式读取
wb = openpyxl.load_workbook('test.xlsx', read_only=True)
# 新建一个工作薄
wb = openpyxl.Workbook()
# 保存xlsx文件
wb.save('test.xlsx')

sheet操作

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 获取工作薄所有的sheet,返回list
sheets = wb.worksheets
# 获取指定sheet页
ws = wb['标记数据']
ws = wb.worksheets[0]
# 获取当前正在使用的sheet页
ws = wb.active
# sheet页属性:表名、最大行数、最大列数
print(ws.title, ws.max_row, ws.max_column)
# 修改sheet名
ws.title = '标记数据2'
# 删除sheet页
wb.remove(ws)
del wb['标记数据']
# 新建sheet页(可以插入到指定的索引处)
wb.create_sheet('test1', 0)

sheet属性

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# 获取行和列,返回tuple
ws[1]
ws['A']
# 对行或列切片,返回tuple
ws[2:4]
ws['A:D']
# 获取所有行或列,返回generator
rows = ws.rows
columns = ws.columns
for row in rows:
for i in row:
print(i.value)

# 添加一行值(添加多行值直接循环添加即可)
ws.append(['人民法院', '故意伤害罪', '89f01654', 5])
# 添加一列值(添加多列值只需将s改为多层list嵌套,其中每一个list为一列)
s = [['NAME', 'Make', 'Tom', 'Smith', 'Suixin']]
for i in zip(*s):
ws.append(i)

cell操作

1
2
3
4
5
6
7
8
# 读取cell
cell = ws['B3']
cell = ws.cell(3, 2) # 先行坐标后列坐标(均从1开始)
# cell属性:所在行、列、坐标、值
print(cell.row, cell.column, cell.coordinate, cell.value)
# 写入cell
cell.value = 'NAME'
ws['B3'] = 'hello'

openpyxl功能全面,还支持:合并单元格、数学运算、单元格格式、迭代器ws.iter_rows()操作等。

使用xlrdxlwtxlutils包处理xls文件

使用xlrd读取xls文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import xlrd

# 打开已有的工作薄
wb = xlrd.open_workbook('test.xls')

# 获取所有的sheet名称
sheet_names = wb.sheet_names()
# 获得sheet对象
ws = wb.sheet_by_index(0)
ws = wb.sheet_by_name('标记数据')
# 获取sheet对象的属性:表名、总行数、总列数
print(ws.name, ws.nrows, ws.ncols)

# 获得某一行或某一列数据,返回list
row_4 = ws.row_values(3)
cloumn_5 = ws.col_values(4)
Row_4 = ws.row(3) # 此方法list中包含单元格类型
Column_5 = ws.col(4) # 此方法list中包含单元格类型

# 获得某一单元格的值
cell_1_1 = ws.cell_value(0, 0)
cell_1_1 = ws.cell(0, 0).value
cell_1_1 = ws.row_values(0)[0]
cell_1_1 = ws.col_values(0)[0]
cell_1_1 = ws.row(0)[0].value
cell_1_1 = ws.col(0)[0].value

xlrd打开为只读模式,不可修改。

使用xlwt写入新建的xls文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import xlwt

# 创建一个新的工作薄
wb = xlwt.Workbook(encoding="utf-8")
# 在其上创建一个新的工作表
ws = wb.add_sheet('S1', cell_overwrite_ok=True)

# 按单元格方式添加数据
ws.write(0, 0, 'HELLO')
# 整行整列的添加数据
title = ['NAME', 'AGE', 'SEX']
for i in range(len(title)):
ws.write(0, i, title[i])
name = ['Tom', 'Smith', 'SuiXin', 'Make']
for j in range(len(name)):
ws.write(j + 1, 0, name[j])
# 保存文件
wb.save('test.xls')

结果:

xlwt写入xls文件

使用xlutils修改xls文件

1
2
3
4
5
6
7
8
9
10
11
12
13
14
import xlrd
from xlutils.copy import copy

rd_book = xlrd.open_workbook('test.xls')
# 使用copy方法将xlrd.Book对象拷贝为一个xlwt.workbook对象,可写入
wt_book = copy(rd_book)
# 获取一个sheet对象(支持index和name)
ws = wt_book.get_sheet('S1')
# 修改工作表
age = [32, 38, 24, 40]
for j in range(len(age)):
ws.write(j + 1, 1, age[j])
# 保存文件
wt_book.save('test_new.xls')

结果:

xlutils修改xls文件

参考

https://juejin.im/entry/5b3b2c7d6fb9a04fe820c5a3
https://blog.csdn.net/sinat_28576553/article/details/81275650
https://zhuanlan.zhihu.com/p/35605282