Python读写操作csv和excle文件代码实例

Python读写操作csv和excle文件代码实例

一、csv文件读写操作

1.1 csv文件读操作

Python中操作csv文件很方便,可以通过csv库来完成读写操作。下面是一个简单的csv文件读取示例:

import csv

with open('data.csv', 'r') as f:

reader = csv.reader(f)

for row in reader:

print(row)

以上代码中使用了 open() 方法打开 csv 文件,并将文件对象传递给 csv.reader 对象,然后使用循环语句遍历整个文件及其内容。每行都被存储为一个列表,可以按索引访问列表中的各个元素。

1.2 csv文件写操作

下面是一个简单的csv文件写入示例:

import csv

with open('data.csv', 'w', newline='') as f:

writer = csv.writer(f)

writer.writerow(['Name', 'Age', 'Gender'])

writer.writerow(['Mike', '25', 'Male'])

writer.writerow(['Lisa', '23', 'Female'])

writer.writerow(['John', '30', 'Male'])

以上代码中,使用了 open() 方法打开 csv 文件,并将文件对象传递给 csv.writer 对象,然后使用 writerow() 方法写入每一行数据。注意,需要在打开文件时指定 newline='' 参数,否则两行数据之间会出现空行。

1.3 csv文件读写操作示例

下面是一个完整的csv文件读写操作示例:

import csv

# csv文件读操作

def read_csv_file(filename):

with open(filename, 'r') as f:

reader = csv.reader(f)

for row in reader:

print(row)

# csv文件写操作

def write_csv_file(filename):

with open(filename, 'w', newline='') as f:

writer = csv.writer(f)

writer.writerow(['Name', 'Age', 'Gender'])

writer.writerow(['Mike', '25', 'Male'])

writer.writerow(['Lisa', '23', 'Female'])

writer.writerow(['John', '30', 'Male'])

# 测试读写csv文件

if __name__ == '__main__':

filename = 'data.csv'

# csv文件读操作

read_csv_file(filename)

# csv文件写操作

write_csv_file(filename)

# 再次读操作,验证写操作结果

read_csv_file(filename)

在上述示例中,定义了两个函数 read_csv_file() 和 write_csv_file() 分别用于csv文件的读操作和写操作,并在主函数中测试了读写csv文件的结果。

二、excle文件读写操作

2.1 excle文件读操作

Python针对excle文件的读写操作主要使用了第三方库openpyxl。下面是一个简单的excle文件读取示例:

import openpyxl

wb = openpyxl.load_workbook('data.xlsx')

sheet = wb.active

for row in sheet.iter_rows(min_row=1):

for cell in row:

print(cell.value)

以上代码中,使用 openpyxl.load_workbook() 方法打开 excle 文件,然后使用 active 属性获取当前的工作表单,使用 iter_rows() 方法遍历整个工作表单。在遍历每个单元格时,使用 cell.value 获取单元格的值。

2.2 excle文件写操作

下面是一个简单的excle文件写入示例:

import openpyxl

wb = openpyxl.Workbook()

sheet = wb.active

sheet['A1'] = 'Name'

sheet['B1'] = 'Age'

sheet['C1'] = 'Gender'

sheet['A2'] = 'Mike'

sheet['B2'] = '25'

sheet['C2'] = 'Male'

sheet['A3'] = 'Lisa'

sheet['B3'] = '23'

sheet['C3'] = 'Female'

wb.save('data.xlsx')

以上代码中,使用 openpyxl.Workbook() 方法创建新的excle文件,并使用 active 属性获取当前的工作表单,然后使用 sheet['A1'] = 'Name' 等方式写入数据,并最后使用 wb.save() 方法保存excle文件。

2.3 excle文件读写操作示例

下面是一个完整的excle文件读写操作示例:

import openpyxl

# excle文件读操作

def read_excle_file(filename):

wb = openpyxl.load_workbook(filename)

sheet = wb.active

for row in sheet.iter_rows(min_row=1):

for cell in row:

print(cell.value)

# excle文件写操作

def write_excle_file(filename):

wb = openpyxl.Workbook()

sheet = wb.active

sheet['A1'] = 'Name'

sheet['B1'] = 'Age'

sheet['C1'] = 'Gender'

sheet['A2'] = 'Mike'

sheet['B2'] = '25'

sheet['C2'] = 'Male'

sheet['A3'] = 'Lisa'

sheet['B3'] = '23'

sheet['C3'] = 'Female'

wb.save(filename)

# 测试读写excle文件

if __name__ == '__main__':

filename = 'data.xlsx'

# excle文件读操作

read_excle_file(filename)

# excle文件写操作

write_excle_file(filename)

# 再次读操作,验证写操作结果

read_excle_file(filename)

在上述示例中,定义了两个函数 read_excle_file() 和 write_excle_file() 分别用于excle文件的读操作和写操作,并在主函数中测试了读写excle文件的结果。

三、总结

以上就是Python对csv和excle文件读写操作的示例代码和讲解。对于csv文件的读写操作,Python自带的csv库已经可以满足绝大部分需求;对于excle文件的读写操作,我们可以使用第三方库openpyxl来完成。

无论是对于csv文件还是excle文件的操作,Python都提供了非常方便和灵活的方式来完成。在实际工作中,我们可以根据自己的需求选择适合自己的方式来完成文件读写操作。

后端开发标签