欢迎访问宙启技术站
智能推送

使用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
电子产品
家具
服装
食品

您可以根据您的实际需求调整函数和用法。例如,如果您的分类数据包含多个列,您可以根据需要扩展写入的行数和写入的列数。