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文件中的类别信息在不同的列或以不同的格式存在,请适当修改代码来适应您的数据。
