Python: Как передать список как параметр функции?

Question

Я только начинаю с Python, и я работаю со следующим кодом, чтобы импортировать файл CSV в sqlite3 таблицу, которую я свободно признаться, я скопировал большую часть из интернета:

with open(getPathTo('my.csv'), 'r') as csvfile: 
    reader = csv.DictReader(csvfile) 
    records = [(row['SEGMENT'], row['Comp 1'], row['Comp 2']) for row in reader] 
c.executemany("INSERT INTO comparison (`SEGMENT`, `Comp 1`, `Comp 2`) VALUES (?,?,?);", records) 
conn.commit() 

Он работает отлично, но я повторяю это для большого количества файлов и таблиц, и я хотел бы превратить его в функцию. Что я стремлюсь к что-то вроде:

def importCSVToTable(file, table, columns) 

Но, учитывая список columns, как я могу использовать его в этой строке:

records = [(row['SEGMENT'], row['Comp 1'], row['Comp 2']) for row in reader] 

Я просто немного Брошенный синтаксис, я думаю.

Answer 1

Вот пример кода, показывающий что-то, что может быть полезно. Мы используем вложенное понимание с оператором фильтрации if, чтобы убедиться, что мы не пытаемся получить доступ к несуществующим элементам dict.

In [3]: def importCSVtoTable(file, table, columns): 
    ...:  # 'mock' data to simulate a reader 
    ...:  reader = [{'SEGMENT': 2, 'Comp 1': 'dogs'}, {'Comp 2': 'cats', 'OTHERTHING': 4} 
    ...:  print [[row[label] for label in columns if label in row] for row in reader] 
    ...: 

In [4]: importCSVtoTable(None, None, ['SEGMENT', 'Comp 1']) 
[[2, 'dogs'], []] 

In [5]: importCSVtoTable(None, None, ['SEGMENT', 'Comp 1', 'Comp 2']) 
[[2, 'dogs'], ['cats']] 

Answer 2

В функции получите список столбцов, а затем для каждого элемента в списке найдите row[list-element] и добавьте в список для записи.

Answer 3

возможно что-то вроде этого? конечно, непроверенный.

def importCSVToTable(conn, filename, table, columns) 
    with open(getPathTo(filename), 'r') as csvfile: 
     reader = csv.DictReader(csvfile) 
     records = [] 
     for row in reader: 
      for col in columns: 
       records.append(row[col]) 
    conn.executemany("INSERT INTO comparison (" + ','.join(columns) + ") VALUES ("+ ','.join(["?"]*columns.length) +");", records) 
    conn.commit() 

Answer 4

Так что я думаю, что вы спрашиваете есть, учитывая список ключей (столбцы), как извлечь их из словаря? Давайте просто использовать файл CSV в памяти, чтобы проверить это:

>>> example_data = """col1,col2,col3\na1,a2,a3\nb1,b2,b3\nc1,c2c3""" 
>>> print example_data 
col1,col2,col 
a1,a2,a3 
b1,b2,b3 
c1,c2c3 

Тогда, если мы имеем CSVReader, основанный на том, что:

>>> import csv, StringIO 
>>> reader = csv.DictReader(StringIO.StringIO(example_data)) 
>>> print reader.fieldnames 
['col1', 'col2', 'col3'] 

Таким образом, если мы хотим перебрать на основе этого списка поля:

>>> for row in reader: 
...  print 'insert into mytable (%s) values (%s)' % (','.join(reader.fieldnames), ','.join(['?']*len(reader.fieldnames))) 
insert into mytable (col1,col2,col3) values (?,?,?) 
insert into mytable (col1,col2,col3) values (?,?,?) 
insert into mytable (col1,col2,col3) values (?,?,?) 

Оттуда, очевидно, вы захотите адаптировать его для своей функции. Но отвечает ли это на ваш вопрос о механизме csvreader и манипулирует списками Python?

(Примечание, это для Python 2.)