Question

格式中有一个csv文件。

x1  0  1  3  6
x2  4  9  4  5
x3  8  6  7  1

我想将它插入一个包含三个字段的数据库＆＃34; x1＆＃34;，＆＃34; x2＆＃34;，＆＃34; x3＆＃34;进入以下数据。

x1|x2|x3
0|4|8
1|9|6
3|10|7
6|5|1

我认为在阅读csv文件时我必须转换数据。

import sqlite3
db="test.sqlite"
con = sqlite3.connect(db)
cur = con.cursor()
dat=open("data","r")
for id,line in enumerate(dat.readlines()):
    ?

con.executemany('insert into test values(?,?,?)', value)

如何在这里获得正确的价值格式？

Answer 1

使用zip(*rows)转置数据，移除第一行并插入.executemany()来电：

import csv
import sqlite3

with open('data', 'r', newline='') as infh:
    rows = list(csv.reader(infh))
    columns = zip(*rows)
    next(columns, None)  # skip the first column

    with sqlite3.connect('test.sqlite') as conn:
        cursor = conn.cursor()
        cursor.executemany('insert into test values (?, ?, ?)', columns)

如何将csv文件中的数据插入sqlite db？

1 个答案: