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

Python代码示例:从CSV文件中读取类别信息并存储

发布时间:2023-12-27 22:19:22

以下是一个示例代码,演示了如何使用Python从CSV文件中读取类别信息并将其存储。

import csv

def read_csv_file(file_name):
    categories = []
    with open(file_name, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            category = row[0]  # 假设类别信息在CSV文件的      列
            categories.append(category)
    return categories

# 读取CSV文件
file_name = 'categories.csv'
categories = read_csv_file(file_name)

# 打印类别信息
for category in categories:
    print(category)

在这个示例代码中,我们首先定义了一个 read_csv_file 函数,它接受一个文件名作为参数,并返回一个包含所有类别信息的列表。该函数通过打开文件并使用CSV读取器来遍历文件的每一行,然后提取类别信息,并将其添加到列表中。

随后我们调用 read_csv_file 函数,并将文件名 'categories.csv' 作为参数传递给它。这会返回一个包含CSV文件中的类别信息的列表。

最后,我们使用一个简单的循环遍历该列表,并将每个类别打印出来。

请注意,示例代码中假设类别信息在CSV文件的 列。如果您的CSV文件中的类别信息在不同的列或以不同的格式存在,请适当修改代码来适应您的数据。