使用Python保存分类到CSV文件的方法:save_categories_to_csv_file()
发布时间:2024-01-11 02:02:21
要将分类保存到CSV文件中,您可以使用Python的csv模块。下面是一个名为save_categories_to_csv_file()的函数,它接受两个参数:categories(分类列表)和filename(要保存的文件名)。
import csv
def save_categories_to_csv_file(categories, filename):
# 创建CSV文件并写入分类数据
with open(filename, mode='w', newline='') as file:
writer = csv.writer(file)
writer.writerow(['Category']) # 写入标题行
for category in categories:
writer.writerow([category]) # 写入每个分类
# 示例用法
categories = ['电子产品', '家具', '服装', '食品']
filename = 'categories.csv'
save_categories_to_csv_file(categories, filename)
在上面的例子中,首先我们导入了csv模块。然后定义了save_categories_to_csv_file()函数。函数内部使用with语句打开文件,使用csv.writer将数据写入文件。首先写入标题行(这里我们只有一个列名为"Category"),然后遍历分类列表,将每个分类写入一行。
接下来,我们提供一个示例用法。定义了一个分类列表和要保存的文件名。然后调用save_categories_to_csv_file()函数以保存分类到CSV文件。
这将创建一个名为categories.csv的文件,并在其中保存分类数据。文件内容如下:
Category 电子产品 家具 服装 食品
您可以根据您的实际需求调整函数和用法。例如,如果您的分类数据包含多个列,您可以根据需要扩展写入的行数和写入的列数。
