python - 如何使用 python 将文本文件导入 SQLite3 表

Question

我目前有一个包含电影 ID、电影名称和流派的 txt 电影文件。例子：

1,Toy Story (1995),Animation|Children's|Comedy
2,Jumanji (1995),Adventure|Children's|Fantasy
3,Grumpier Old Men (1995),Comedy|Romance
4,Waiting to Exhale (1995),Comedy|Drama
5,Father of the Bride Part II (1995),Comedy

我正在尝试将此文件导入使用 Python 创建的 SQLite3 表中：

import sqlite3

conn = sqlite3.connect('CSC 452 Final.db')
c=conn.cursor()

删除表

c.execute('''DROP TABLE IF EXISTS temp_movie''')
print("Table temp_movie dropped.")

为电影创建表

c.execute('''CREATE TABLE temp_movie (
    MovieID          NUMERIC (5),
    movie_title_year VARCHAR2 (255),
    movie_categories VARCHAR2 (255),
    CONSTRAINT movies_movieID_PK PRIMARY KEY (
        MovieID
    ))''')
print("Table temp_movie successfully created.")
conn.commit()

将我的文本文件导入此数据库的最佳方法是什么，分隔符为“，”？

标签： pythonsqlite

Answer 1

您可以读取txt文件并拆分,：

file_data = [i.strip('\n').split(',') for i in open('filename.txt')]
new_data = [[int(a), *b] for a, *b in file_data] #convert string id to integer
conn = sqlite3.connect('CSC 452 Final.db')
c = conn.cursor()
c.executemany('INSERT INTO temp_movie VALUES (?, ?, ?)', new_data)