【python】:如何把csv文件做成一个变量?

试想,一个项目的测试用例,对应多个csv文件。但是这种方法只能读取到指定的一个Address.csv文件,怎么办?
--------把Address.csv做成一个变量。接受一个文件名,通过文件名找文件。

完整代码:

#coding=utf-8
import csv
import os

def read(filename):
    base_path=os.path.dirname(__file__) # 获取当前文件目录
    path=base_path.replace('py_file','data/'+filename)# 目录替换

    with open(path) as file:
        data=csv.reader(file)
        list=[]
        i=0
        for row in data:
            if i==0:
                pass
            else:
                list.append(row)
            i=i+1
    return list
if __name__ == '__main__':# main以下的代码只会在当前文件中执行,不可被其他的方法调用
    list=read("Address.csv")
    print(list)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值