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

使用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文件中的类别信息。