当前位置:首页python > 正文

python获取csv文件中某一列

作者:野牛程序员:2023-07-24 07:42:40python阅读 2736

在Python中,可以使用csv模块来读取CSV文件并获取其中的某一列数据。下面是一个简单的示例代码,假设要获取名为"column_name"的列:

import csv

def get_column_from_csv(csv_file, column_name):
    column_data = []
    with open(csv_file, 'r', newline='') as file:
        reader = csv.DictReader(file)
        for row in reader:
            if column_name in row:
                column_data.append(row[column_name])
    return column_data

# 用法示例
csv_file_path = 'path/to/your/csv_file.csv'
column_name_to_get = 'column_name'
column_data = get_column_from_csv(csv_file_path, column_name_to_get)

print(column_data)

确保将"path/to/your/csv_file.csv"替换为实际的CSV文件路径,"column_name"替换为想要提取的列的名称。这个函数会返回一个包含该列所有数据的列表。

请注意,此代码假设CSV文件的第一行包含列名。如果CSV文件没有列名行,可以使用csv.reader而不是csv.DictReader来处理文件。


野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击