使用Python编写的load_categories_from_csv_file()函数加载CSV文件中的类别信息
发布时间:2023-12-27 22:17:32
下面是使用Python编写的load_categories_from_csv_file()函数的实现:
import csv
def load_categories_from_csv_file(filename):
categories = []
with open(filename, 'r') as file:
reader = csv.reader(file)
for row in reader:
categories.append(row[0])
return categories
这个函数通过打开给定的CSV文件,使用csv.reader()函数读取文件内容,并将每一行的 个数据作为类别信息添加到一个列表中。最后,函数返回包含所有类别信息的列表。
使用例子:
假设我们有一个名为categories.csv的CSV文件,其中包含以下类别信息:
Category 1 Category 2 Category 3 Category 4 Category 5
我们可以使用load_categories_from_csv_file()函数加载这些类别信息,并打印出来:
filename = 'categories.csv' categories = load_categories_from_csv_file(filename) print(categories)
输出结果应该是:
['Category 1', 'Category 2', 'Category 3', 'Category 4', 'Category 5']
这样,我们就成功地加载了CSV文件中的类别信息。
