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

- 上一篇:python获取csv文件中某一行
- 下一篇:python读取csv中url