如何在Python中使用polars包读取SQLite数据库文件

Question

我想使用 package.json 读取 SQLite 数据库文件（database.sqlite）polars。我尝试以下操作未成功：

import sqlite3
import polars as pl

conn = sqlite3.connect('database.sqlite')
df = pl.read_sql("SELECT * from table_name", conn)

print(df)

出现以下错误：

AttributeError: 'sqlite3.Connection' object has no attribute 'split'

有什么建议么？

Answer 1

从文档中，您可以看到 pl.read_sql 接受连接字符串作为参数，并且您正在发送对象 sqlite3.Connection，这就是您收到该消息的原因。

您应该首先生成连接字符串，这是您的数据库的 url

db_path = 'database.sqlite'
connection_string = 'sqlite://' + db_path

之后，您可以输入更新的下一行，这给您带来了问题：

df = pl.read_sql("SELECT * from table_name", connection_string)