当前位置:首页python > 正文

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 文件的路径,column1column2 是要读取的两列的列名。函数使用 csv.reader 对象逐行读取 CSV 文件,通过 header.index 查找列索引,并将每一行的对应列的值添加到 data 列表中。

你需要将 filename 替换为你实际的 CSV 文件路径,并提供正确的列名作为 column1column2 的值。执行函数后,它会返回一个包含所选两列数据的列表。可以根据需要对返回的数据进行进一步处理或输出。


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

最新推荐

热门点击