Python 操作csv和excel表格

时间：2023-03-08 18:35:21

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') ＃　保存文件

相关文章