网站首页 > 文章精选 正文
引言:
代码实现了功能:读取 CSV 文件、获取指定单元格的值、写入 CSV 文件、追加数据到 CSV 文件、删除 CSV 文件
代码封装如下:
import csv
import os
class CSVHandler:
def __init__(self, file_path):
"""
初始化CSVHandler对象
:param file_path: csv文件的路径
"""
self.file_path = file_path #todo 存储文件路径
def read_csv(self):
data = [] #todo 用于存储csv文件的所有行数据
try:
with open(self.file_path, 'r', newline='', encoding='utf-8') as file:
reader = csv.reader(file)
#todo 逐行读取csv内容,并将每一行数据添加到data列表中
for row in reader:
data.append(row)
#todo 打印读取的所有数据(可选)
for row in data:
print(row)
return data #todo 返回读取到的所有数据
except FileNotFoundError:
print(f"文件 '{self.file_path}' 未找到")
except Exception as e:
print(f"读取文件时发生错误: {e}")
def get_csv_cell_value(self, row_number, col_letter):
try:
with open(self.file_path, 'r', newline='', encoding='utf-8') as file:
reader = csv.reader(file)
data = list(reader) #todo 将csv文件内容读取为列表
row_index = row_number - 1 #todo 将行号转换为列表的索引(Python索引从0开始)
col_index = ord(col_letter.lower()) - 97 #todo 将列字母转换为对应的索引,'A' -> 0, 'B' -> 1, 等等
if row_index < 0 or row_index >= len(data):
print(f"行号 {row_number} 超出了有效范围")
return None
if col_index < 0 or col_index >= len(data[row_index]):
print(f"列字母 {col_letter} 超出了有效范围")
return None
return data[row_index][col_index] #todo 返回指定单元格的值
except FileNotFoundError:
print(f"文件 '{self.file_path}' 未找到")
except Exception as e:
print(f"获取单元格值时发生错误: {e}")
def write_csv(self, data):
try:
with open(self.file_path, 'w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
#todo 将传入的数据逐行写入csv文件
for row in data:
writer.writerow(row)
print("数据已成功写入文件")
except Exception as e:
print(f"写入文件时发生错误: {e}")
def append_csv(self, data):
try:
with open(self.file_path, 'a', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
#todo 将传入的数据逐行追加到csv文件
for row in data:
writer.writerow(row)
print("数据已成功追加到文件")
except Exception as e:
print(f"追加数据时发生错误: {e}")
def delete_csv(self):
try:
if os.path.exists(self.file_path):
os.remove(self.file_path) #todo 删除文件
print(f"文件 '{self.file_path}' 已删除")
else:
print(f"文件 '{self.file_path}' 不存在")
except Exception as e:
print(f"删除文件时发生错误: {e}")
if __name__ == '__main__':
#todo 示例代码:以下代码将执行CSV操作
file_path = r"D:\1.csv" #todo 设置文件路径
csv_handler = CSVHandler(file_path) #todo 创建CSVHandler对象
#todo 读取csv文件
csv_handler.read_csv()
#todo 获取指定单元格的值(例如第二行,第三列,列标为'C')
cell_value = csv_handler.get_csv_cell_value(row_number=2, col_letter='C')
if cell_value is not None:
print(f"指定单元格的值是: {cell_value}")
#todo 写入数据(覆盖现有文件内容)
data_to_write = [["Name", "Age", "City"], ["Alice", 30, "New York"], ["Bob", 25, "Los Angeles"]]
csv_handler.write_csv(data_to_write)
#todo 追加数据到文件末尾
data_to_append = [["张三", 35, "上海"], ["李四", 40, "深圳"]]
csv_handler.append_csv(data_to_append)
#todo 删除csv文件
#todo csv_handler.delete_csv()
注意:函数csv_handler.delete_csv()此操作将永久删除文件,使用时请小心
- 上一篇: Python数据类型之列表(List)类型
- 下一篇: Python中的列表和元组,你了解多少?
猜你喜欢
- 2025-01-16 一文了解 Python 列表
- 2025-01-16 Python学习笔记——列表
- 2025-01-16 Python中list列表函数用法大全(思维脑图加详细解读)
- 2025-01-16 Python教程:列表的排序操作
- 2025-01-16 2 常见的Python数据结构-元组、列表
- 2025-01-16 一篇文章带你弄懂Python基础之列表介绍和循环遍历
- 2025-01-16 Python之容器:列表是个百宝箱,什么都能往里装
- 2025-01-16 Python 列表生成式全解
- 2025-01-16 python散装笔记——17: 数组
- 2025-01-16 Python中的列表和元组,你了解多少?
- 最近发表
- 标签列表
-
- newcoder (56)
- 字符串的长度是指 (45)
- drawcontours()参数说明 (60)
- unsignedshortint (59)
- postman并发请求 (47)
- python列表删除 (50)
- 左程云什么水平 (56)
- 计算机网络的拓扑结构是指() (45)
- 稳压管的稳压区是工作在什么区 (45)
- 编程题 (64)
- postgresql默认端口 (66)
- 数据库的概念模型独立于 (48)
- 产生系统死锁的原因可能是由于 (51)
- 数据库中只存放视图的 (62)
- 在vi中退出不保存的命令是 (53)
- 哪个命令可以将普通用户转换成超级用户 (49)
- noscript标签的作用 (48)
- 联合利华网申 (49)
- swagger和postman (46)
- 结构化程序设计主要强调 (53)
- 172.1 (57)
- apipostwebsocket (47)
- 唯品会后台 (61)
- 简历助手 (56)
- offshow (61)