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

Python中使用load_categories_from_csv_file()函数从CSV文件中加载类别信息

发布时间:2023-12-23 10:13:13

在Python中,可以使用load_categories_from_csv_file()函数从CSV文件中加载类别信息。该函数将读取CSV文件,并返回一个包含类别信息的列表。

以下是一个使用load_categories_from_csv_file()函数从CSV文件中加载类别信息的示例:

import csv

def load_categories_from_csv_file(file_path):
    categories = []
    with open(file_path, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            categories.append(row[0])
    return categories

# 示例CSV文件内容:
# category1
# category2
# category3

file_path = 'categories.csv'
loaded_categories = load_categories_from_csv_file(file_path)

print(loaded_categories)

上述代码中,首先定义了一个load_categories_from_csv_file()函数,该函数接受一个CSV文件的路径作为参数。在函数内部,打开指定的CSV文件,并使用csv.reader()函数创建一个CSV读取器。

然后,通过遍历读取器的每一行,将每行的 个元素(即类别信息)添加到一个名为categories的列表中。

最后,函数返回包含所有类别信息的列表。

在示例代码的最后部分,指定了一个CSV文件路径,并调用load_categories_from_csv_file()函数,将加载的类别信息保存在loaded_categories变量中。

最后,打印loaded_categories的值,即输出了从CSV文件中加载的类别信息。

这是一个简单的使用load_categories_from_csv_file()函数从CSV文件中加载类别信息的例子。你可以根据实际需求进行适当的修改和扩展。