保存分类到CSV文件:使用Python的save_categories_to_csv_file()函数
发布时间:2024-01-11 02:01:41
要将分类保存到CSV文件中,可以使用Python的CSV模块。下面是一个示例函数save_categories_to_csv_file(categories, filename),用于将分类保存到CSV文件中。
import csv
def save_categories_to_csv_file(categories, filename):
# 创建CSV文件并打开,使用 'w' 模式表示写入操作
with open(filename, 'w', newline='') as file:
# 创建一个CSV写入器
writer = csv.writer(file)
# 写入标题行
writer.writerow(['Category'])
# 写入每个分类
for category in categories:
writer.writerow([category])
以上函数接受两个参数:
- categories:一个包含分类名称的列表
- filename:要保存的CSV文件的文件名
函数将创建一个CSV文件并写入分类。文件中的每行将只包含一个分类名称。
下面是一个使用例子:
categories = ['分类1', '分类2', '分类3', '分类4'] save_categories_to_csv_file(categories, 'categories.csv')
在上面的例子中,我们假设已经定义了分类列表categories,然后调用save_categories_to_csv_file()函数将分类保存到名为categories.csv的CSV文件中。
CSV文件的内容将如下所示:
Category 分类1 分类2 分类3 分类4
这样,我们就成功将分类保存到了CSV文件中。可以根据实际需求进行修改,例如修改标题行名称或添加更多的分类属性。
