python 定义函数(modify_ booktype()):修改图书类别信息,将图书类别信息从文件(booktype.csv)中修改
时间: 2024-01-09 11:03:21 浏览: 151
好的,下面是一个简单的实现:
```python
import csv
def modify_booktype(booktype_id, new_booktype):
"""
修改图书类别信息,将图书类别信息从文件(booktype.csv)中修改
:param booktype_id: 要修改的图书类别 ID
:param new_booktype: 新的图书类别名称
"""
with open('booktype.csv', 'r', encoding='utf-8') as f:
reader = csv.reader(f)
rows = list(reader)
# 查找要修改的图书类别
found = False
for i in range(len(rows)):
if rows[i][0] == booktype_id:
rows[i][1] = new_booktype
found = True
break
# 如果找到了要修改的图书类别,则写回文件
if found:
with open('booktype.csv', 'w', encoding='utf-8', newline='') as f:
writer = csv.writer(f)
writer.writerows(rows)
print(f'图书类别 {booktype_id} 修改成功!')
else:
print(f'找不到图书类别 {booktype_id}!')
```
这个函数首先打开 `booktype.csv` 文件,读取所有行,并将它们存储在一个列表 `rows` 中。然后,它遍历 `rows` 列表,查找要修改的图书类别。如果找到了该图书类别,则将其名称修改为 `new_booktype`。最后,将修改后的列表写回 `booktype.csv` 文件中。
需要注意的是,这个函数假设 `booktype.csv` 文件的第一列是图书类别的 ID,第二列是图书类别的名称。如果文件的格式不同,需要根据实际情况进行修改。此外,这个函数还假设 `booktype.csv` 文件使用 UTF-8 编码,如果文件使用其它编码,也需要进行相应的修改。
阅读全文