1

我有一个 csv 文件“users.csv”,如下所示:

AppID,Perm,LastLogon,Email,FirstName
jimmeny1,admin,today,jim5@zerlina.com,pop1
jimmeny2,admin,today,jim10@zerlina.com,pop2
jimmeny3,admin,today,jim6@zerlina.com,pop3

我创建了以下函数以将任意数量的列从 csv 加载到 sqlite 表中:

import sqlite3, csv

def bulkload(file, **args):
    sqlite = sqlite3.connect('mydb.db')
    s1 = sqlite.cursor()
    cols_string = '('
    cols = []
    for i in args.iteritems():
        col = '{} {}'.format(i[0], i[1][1])
        cols.append(col)
    for x, i in enumerate(cols):
        if x != len(cols) - 1:
            cols_string += i + ','
        if x == len(cols) - 1:
            cols_string += i + ')'
    sql1 = "CREATE TABLE IF NOT EXISTS mytable {}".format(cols_string)
    s1.execute(sql1)

对函数的示例调用:

bulkload('users.csv', appid=(0, 'text'), perm=(1,'text'), email=(3,'text'))

是否有更简单、更优雅或更简单的 Pythonic 方式来创建声明列名和存储类的字符串“cols_string”?

4

1 回答 1

1

您应该替换以下内容

cols_string = '('
for x, i in enumerate(cols):
    if x != len(cols) - 1:
        cols_string += i + ','
    if x == len(cols) - 1:
        cols_string += i + ')'

使用以下内容,这也更加pythonic和更快(由于 ''.join() 方法)

cols_string = '(' + ','.join(cols) + ')'
于 2013-11-15T07:42:26.470 回答