Python中的load_categories_from_csv_file()函数用于加载CSV文件中的类别列表
发布时间:2023-12-23 10:15:41
load_categories_from_csv_file()函数用于从CSV文件中加载类别列表。
以下是一个使用例子:
假设有一个名为categories.csv的CSV文件,其内容如下:
id,category 1,汽车 2,电子产品 3,家居 4,食品
我们可以使用load_categories_from_csv_file()函数加载该文件,并将类别返回为一个列表。代码如下:
import csv
def load_categories_from_csv_file(file_path):
categories = []
with open(file_path, mode='r') as file:
reader = csv.DictReader(file)
for row in reader:
categories.append(row['category'])
return categories
file_path = 'categories.csv'
categories = load_categories_from_csv_file(file_path)
print(categories)
运行上述代码,输出结果为:
['汽车', '电子产品', '家居', '食品']
上述代码解析如下:
1. 首先,我们导入了csv模块,以便处理CSV文件。
2. 接下来,定义了一个load_categories_from_csv_file()函数,该函数接受一个文件路径作为参数。
3. 在函数内部,我们创建了一个空列表categories,用于存储从CSV文件中提取的类别。
4. 然后,我们使用open()函数打开CSV文件,并指定文件路径和'r'模式(以只读方式打开文件)。
5. 通过使用csv.DictReader()函数,我们将打开的文件对象传递给reader,以便处理CSV数据。
6. 接下来,我们使用一个for循环遍历reader中的每一行数据。
7. 在循环的每一次迭代中,我们将当前行的'category'字段的值提取出来,并将其添加到categories列表中。
8. 一旦遍历完所有行,我们关闭文件。
9. 最后,我们将categories列表作为函数的返回值。
这样,我们就可以从CSV文件中加载类别列表,并将其作为一个Python列表使用。
