这个python脚本可以缩短/优化,怎么样？

Question

我不习惯以python的方式做事,但我几乎可以肯定以下脚本可以压缩.我不是在寻找速度优化,我正在寻找更易读的代码.让我所关心的所有内容都变慢,但有什么方法可以让它看起来更像Python.

我只是在一个填充了zipcodes的csv文件中读取并填充了一个包含23列的数据库.很多打字.我也不喜欢像行的索引这样的硬编码.此外,multis ='%s,'*23后跟multis = multis [: - 2]感觉很脏.

期待看到你们中的一些人能够想出什么.

#!/usr/bin/python

import csv
import MySQLdb

db = MySQLdb.connect(host="localhost", user="root", db="test")
c = db.cursor()
f_csv = 'zip-codes-database-STANDARD.csv'
csvReader = csv.reader(open(f_csv))
ziplist = []
multis = '%s, '*23
multis = multis[:-2]
for row in csvReader:
    c.execute("""INSERT INTO lock_zipcodes_complete 
            (`zipcode`, `city`, `state`, `county`, `areacode`,
            `citytype`, `city_alias_abbreviation`,
            `city_alias_name`, `latitude`, `longitude`, `timezone`,
            `elevation`, `county_fips`, `dst`,
            `preferred_last_line_key`, `classification_code`,
            `multicounty`, `state_fips`, `city_state_key`,
            `city_alias_code`, `primary_record`, `city_mixed_case`,
            `city_alias_mixed_case`) VALUES(""" + multis + ')',
            (row[0], row[1], row[2], row[3], row[4], row[5], row[6],
            row[7], row[8], row[9], row[10], row[11], row[12],
            row[13], row[14], row[15], row[16], row[17], row[18],
            row[19], row[20], row[21], row[22]))

Answer 1

这部分:

multis = '%s, '*23
multis = multis[:-2]

应该

multis = ', '.join(['%s'] * 23)

ziplist 未使用,因此您可以删除设置它的行.

            (row[0], row[1], row[2], row[3], row[4], row[5], row[6],
            row[7], row[8], row[9], row[10], row[11], row[12],
            row[13], row[14], row[15], row[16], row[17], row[18],
            row[19], row[20], row[21], row[22])

应该只是tuple(row)(对于你使用它的具体方式,只row会这样做).

明确的列名列表INSERT是令人不快的,但如果您不确定该数据库表中的所有列是否正确(或者当然,如果您确定不是这种情况),则需要; 但这是一个SQL - vs - CSV"阻碍不匹配"问题,而不是Python一个;-).

整个循环:

for row in csvReader:
  c.execute(sql_statement, row)

可以选择进一步压缩到:

c.executemany(sql_statement, csvReader)