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

s71maibg 于 2023-02-05 发布在 SQLite

关注(0)|答案(1)|浏览(445)

我想使用polars包读取SQLite数据库文件（database.sqlite）。我尝试了以下操作，但没有成功：

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'

有什么建议吗？

1条答案

在docs中，您可以看到pl.read接受连接字符串作为参数，并且您正在发送对象sqlite3.Connection，这就是您得到该消息的原因。
您应该首先生成连接字符串，它是数据库的url

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

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

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