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都提供了非常方便和灵活的方式来完成。在实际工作中,我们可以根据自己的需求选择适合自己的方式来完成文件读写操作。