python读取csv某两列
作者:野牛程序员:2023-07-19 08:09:03python阅读 2799
要读取 CSV 文件中的某两列,可以使用 Python 的内置模块 csv
来实现。下面是一个简单的示例代码,演示如何读取 CSV 文件的特定列:
import csv def read_csv_columns(filename, column1, column2): with open(filename, 'r') as csvfile: reader = csv.reader(csvfile) header = next(reader) # 获取标题行 # 确定列索引 try: index1 = header.index(column1) index2 = header.index(column2) except ValueError: print("指定的列名不存在") return data = [] for row in reader: if len(row) > max(index1, index2): data.append((row[index1], row[index2])) return data # 示例用法 filename = 'data.csv' column1 = 'Column1' column2 = 'Column2' result = read_csv_columns(filename, column1, column2) for row in result: print(row)
在上述代码中,read_csv_columns
函数接受三个参数:filename
是 CSV 文件的路径,column1
和 column2
是要读取的两列的列名。函数使用 csv.reader
对象逐行读取 CSV 文件,通过 header.index
查找列索引,并将每一行的对应列的值添加到 data
列表中。
你需要将 filename
替换为你实际的 CSV 文件路径,并提供正确的列名作为 column1
和 column2
的值。执行函数后,它会返回一个包含所选两列数据的列表。可以根据需要对返回的数据进行进一步处理或输出。
野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892

- 上一篇:python删除csv某一行
- 下一篇:python csv文件