python读取csv某两列
作者:野牛程序员:2023-07-19 08:09:03python阅读 2969
要读取 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文件
