Python 操作csv和excel表格
1. 操作csv表格
使用的库 csv
1. csv文件里读取数据
代码
1. 以列表形式打开
import csv
f = open('csv_test.csv', 'r') # 打开csv文件
csv_reader = csv.reader(f) # 将打开的文件装换成csv可读的对象
for each in csv_reader: # 打印,结果是个列表
print(each)
f.close()
结果
['name', 'age', 'score']
['张三', '18', '100']
['王二', '17', '98']
['麻子', '20', '99']
2.以字典形式打开
import csv
f = open('csv_test.csv', 'r') # 打开csv文件
csv_reader = csv.DictReader(f) # 将打开的文件装换成csv可读的对象,注意是DictReader()
for each in csv_reader:
print(each['name'])
f.close()
结果
张三
王二
麻子
说明
不能直接print(csv_reader) 结果为
<_csv.reader object at 0x7f67f2328198>(第一种)
<csv.DictReader object at 0x7f29ef0a4710>(第二种)
以字典形式打开会将表头作为字典的key,而其余的每一行会成为字典的value
2. 将数据写进csv文件
代码
1.以列表的方式
import csv
contents = [['id', 'name', 'hobby'],
['001', '老王', 'Java'],
['002','老胡','Python'],
['003','老李','PHP']]
with open('csv_test2.csv', 'a', newline='') as f:
csv_writer = csv.writer(f)
for each in contents:
csv_writer.writerow(each)
结果
| id | name | hobby |
|---|---|---|
| 1 | 老王 | Java |
| 2 | 老胡 | Python |
| 3 | 老李 | PHP |
2.以字典的方式
import csv
header = ['id', 'name', 'hobby']
contents = [{'id':'1','name':'老王','hobby':'Java'},
{'id':'2','name':'老胡','hobby':'Python'},
{'id':'3','name':'老李','hobby':'PHP'}]
with open('csv_test4.csv', 'a', newline='') as f:
csv_writer = csv.DictWriter(f, header)
csv_writer.writeheader()
for each in contents:
csv_writer.writerow(each)
结果
| id | name | hobby |
|---|---|---|
| 1 | 老王 | Java |
| 2 | 老胡 | Python |
| 3 | 老李 | PHP |
说明
打开新的文件时需要newline='',否则会产生空行
上面的代码也可以用csv_writer.writerows(contents)实现(两种均可)
以字典形式写入时必须先写头具体操作见代码
2. 操作excel表格
使用的库 xlrd(读), xlwt(写)
1. 从excel文件里读取数据
import xlrd
book = xlrd.open_workbook('excel_test1.xls') # 打开excel文件
sheet_1 = book.sheets()[0] # 打开第一张表
num_rows = sheet_1.nrows # 获取当前表的所有的行数
num_cols = sheet_1.ncols # 获取当前表的所有的列数
# 遍历行, 会将每一行以列表的形式输出
for row in range(num_rows):
row_values = sheet_1.row_values(row) # 使用的函数是row_values()
print(row_values)
# 遍历列, 会将每一列以列表的形式输出
for col in range(num_cols):
col_values = sheet_1.col_values(col) # 使用的函数是col_values()
print(col_values)
# 遍历单元格, 从上到下从左到右输出每一个单元格的内容
for row in range(num_rows):
for col in range(num_cols):
cell_value = sheet_1.cell_value(row, col) # 使用的函数是cell_value()
print(cell_value)
说明
主要函数,属性
book = xlrd.open_workbook('excel_test1.xls') # 打开excel文件
sheet_1 = book.sheets()[0] # 打开第一张表num_rows = sheet_1.nrows # 获取当前表的所有的行数
num_cols = sheet_1.ncols # 获取当前表的所有的列数row_values = sheet_1.row_values(row) # 使用的函数是row_values()
col_values = sheet_1.col_values(col) # 使用的函数是col_values()
cell_value = sheet_1.cell_value(row, col) # 使用的函数是cell_value()
注意xlrd不可以改动excel文件
2. 将数据写进excel文件中
import xlwt
# 创建Workbook对象
book = xlwt.Workbook(encoding='utf-8', style_compression=0)
'''
Workbook类初始化时有encoding和style_compression参数
encoding='utf-8',为输入中文
style_compression,表示是否压缩
'''
# 添加表格
sheet = book.add_sheet('sheet1', cell_overwrite_ok=True)
'''
sheet1为表格的名称
cell_overwrite_ok,表示是否可以覆盖单元格
'''
# 向表sheet1中添加数据
sheet.write(0, 0, 'id') # 前两个参数为单元格坐标,这里表示第一行第一列(从0开始),后面为内容
sheet.write(0,1, 'name')
sheet.write(1,0, '1')
sheet.write(1,1, '老胡')
# 保存到指定的Excel文件中
book.save('excel_test2.xls')
说明
主要函数
book = xlwt.Workbook(encoding='utf-8', style_compression=0) # 创建Workbook对象
sheet = book.add_sheet('sheet1', cell_overwrite_ok=True) # 添加表格
sheet.write(0, 0, 'id') # 前两个参数为单元格坐标,这里表示第一行第一列(从0开始),后面为内容
book.save('excel_test2.xls') ＃ 保存文件
