利用Python中的load_categories_from_csv_file()函数导入CSV文件中的类别
发布时间:2023-12-23 10:12:49
Python中的load_categories_from_csv_file()函数用于从CSV文件中导入类别。该函数需要一个参数,即CSV文件的路径。它会读取CSV文件,并将其中的类别返回作为一个列表。以下是一个使用例子,具体步骤如下:
首先,导入所需的模块,包括csv模块和自定义函数load_categories_from_csv_file()。
import csv
def load_categories_from_csv_file(file_path):
with open(file_path, 'r') as file:
reader = csv.reader(file)
categories = []
for row in reader:
categories.append(row[0])
return categories
然后,调用load_categories_from_csv_file()函数,并传入CSV文件的路径作为参数。
file_path = 'categories.csv' categories = load_categories_from_csv_file(file_path)
假设CSV文件的内容如下所示:
Category Sports Technology Fashion
load_categories_from_csv_file()函数将读取该文件,并返回一个包含类别的列表:
print(categories)
输出结果为:['Sports', 'Technology', 'Fashion']。
这样,我们就成功地从CSV文件中导入了类别,并将其存储在一个列表中供后续使用。你可以根据需要在列表上执行不同的操作,比如打印每个类别或者进行其他数据处理。
