使用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文件。根据需要,你可以修改函数来适应特定的数据结构、文件名和列名。
