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文件中加载类别信息的例子。你可以根据实际需求进行适当的修改和扩展。
