使用Python的load_categories_from_csv_file()函数从CSV文件中读取类别数据
发布时间:2023-12-23 10:15:18
Python中可以使用pandas库来读取CSV文件。首先,我们需要安装pandas库。可以使用以下命令来安装:
pip install pandas
安装完成后,可以使用以下代码来读取CSV文件并加载类别数据:
import pandas as pd
def load_categories_from_csv_file(file_path):
# 读取CSV文件
df = pd.read_csv(file_path)
# 提取类别列
categories = df['Category'].tolist()
return categories
在上述代码中,我们定义了一个名为load_categories_from_csv_file的函数,该函数接受一个file_path参数,表示要读取的CSV文件的路径。
函数中的 步是使用pd.read_csv()函数读取CSV文件,该函数返回一个DataFrame对象,表示整个CSV文件。
接下来,我们使用df['Category']来提取CSV文件中的“Category”列。然后,我们使用.tolist()将该列转换为Python的列表对象。
最后,我们返回提取的类别数据。
以下是一个使用load_categories_from_csv_file()函数的例子:
file_path = 'categories.csv' categories = load_categories_from_csv_file(file_path) print(categories)
假设CSV文件的内容如下:
Category Cat Dog Bird Fish
运行上述代码,将输出:
['Cat', 'Dog', 'Bird', 'Fish']
这样,我们就成功地从CSV文件中读取了类别数据。
