如何在 Python 中使用 polars package 讀取 SQLite 數據庫文件

Question

我想使用polars package 讀取一個 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'

有什么建議么？

Answer 1

從文檔中，您可以看到 pl.read_sql 接受連接字符串作為參數，並且您正在發送 object sqlite3.Connection，這就是您收到該消息的原因。

您應該首先為您的數據庫生成連接字符串，即 url

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

之后，您可以輸入更新的下一行，這給您帶來了問題：

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