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

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列表使用。