根据行将CSV数据存储到单独的对象中

时间:2017-12-21 11:17:33

标签: python csv selenium

我正在尝试让我的Python / Selenium脚本读取一个包含5列和可变行数的csv,然后将每行的列数据存储在单独的对象中。

基本上我希望能够使用CSV中的数据在稍后在脚本中调用的函数中使用。

免责声明 - 我是一个完整的菜鸟,他只学习了2周的Python,所以如果这是一个愚蠢的问题,请道歉。

到目前为止,这是我的代码:

def csv_read(file_name):
    open_file = open(file_name,encoding="utf-8-sig")
    reader = csv.reader(open_file)
    my_list = []
    for row in reader:
            my_list.append(myClass(row[0], row[1], row[2], row[3], row[4], row[5])) 

由于

1 个答案:

答案 0 :(得分:1)

访问的列太多了。您说CSV文件包含5列,row[5]是第6列,因为索引从0开始。除此之外,还有什么问题?

它似乎应该可以工作,即my_list将是myClass类实例的列表。您可以通过迭代my_list

来调用这些类的方法
for x in my_list:
    x.some_method()

其中some_method()myClass中定义的方法。

或者您可以将实例传递给另一个处理它们的函数:

for x in my_list:
    some_function(x)