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

使用Python编写的save_categories_to_csv_file()函数:将分类信息保存为CSV文件

发布时间:2024-01-11 02:05:51

下面是使用Python编写的save_categories_to_csv_file()函数,该函数将分类信息保存为CSV文件。

import csv

def save_categories_to_csv_file(categories, file_name):
    # 创建CSV文件
    with open(file_name, mode='w', newline='') as csv_file:
        writer = csv.writer(csv_file)
        # 写入CSV文件的列名
        writer.writerow(['Category ID', 'Category Name'])
        
        # 写入分类信息
        for category in categories:
            writer.writerow([category['id'], category['name']])

使用例子:

categories = [
    {'id': 1, 'name': 'Sports'},
    {'id': 2, 'name': 'Technology'},
    {'id': 3, 'name': 'Fashion'},
    {'id': 4, 'name': 'Food'}
]

save_categories_to_csv_file(categories, 'categories.csv')

在上面的例子中,我们定义了一个包含了四个分类信息的列表categories。然后我们调用save_categories_to_csv_file()函数,将该列表保存为名为categories.csv的CSV文件。

该函数首先打开一个CSV文件,并将文件句柄存储在csv_file中。然后,我们创建一个csv.writer对象,用于将数据写入CSV文件。

我们首先使用writer.writerow()函数写入CSV文件的列名,列名为['Category ID', 'Category Name']

接下来,我们使用一个循环遍历分类列表categories,并使用writer.writerow()函数将分类的ID和名称写入CSV文件中。

最后,我们关闭文件句柄,完成保存。

以上是一个简单的Python函数,可以将分类信息保存为CSV文件。根据需要,你可以修改函数来适应特定的数据结构、文件名和列名。