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

Python中加载类别信息的CSV文件读取方法

发布时间:2023-12-27 22:18:58

在Python中,可以使用csv模块来读取CSV文件。CSV文件是以逗号作为字段分隔符的文本文件,常用于存储表格数据。

要读取CSV文件中的类别信息,可以按照以下步骤进行操作:

1. 导入csv模块:

import csv

2. 打开CSV文件:

with open('categories.csv', 'r') as file:
    csv_reader = csv.reader(file)

在这个例子中,我们打开了名为categories.csv的CSV文件,并创建了一个csv.reader对象csv_reader来读取文件内容。

3. 读取CSV文件中的类别信息:

categories = []
for row in csv_reader:
    categories.append(row[0])

在这个例子中,我们创建了一个空列表categories来存储读取到的类别信息。通过循环遍历csv_reader对象的每一行,我们可以按需求提取每一行的类别信息。在这个例子中,我们将每一行中的 个字段(索引为0)作为类别信息,并将其添加到categories列表中。

完整的例子代码如下所示:

import csv

categories = []
with open('categories.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        categories.append(row[0])
        
print(categories)

假设categories.csv文件的内容如下所示:

category1
category2
category3

上述代码执行后的输出结果将是['category1', 'category2', 'category3'],即将CSV文件中的类别信息读取到了categories列表中。

需要注意的是,上述代码中将所有的类别信息都存储在了一个列表categories中。如果你想要以更适合你的应用程序需求的方式存储类别信息,你可以根据需要进行相应的修改。