如何在Python中使用Polars向SQLite数据库写入数据

发布于02月29日

import polars as pl
import sqlite3

conn = sqlite3.connect("test.db")
df = pl.DataFrame({"col1": [1, 2, 3]})

根据pl.write_database的文档，我需要为SQLite数据库传递一个连接URI字符串，例如"SQLite:/Path/to/lobase.db":

df.write_database("test_table", f"sqlite:////test.db", if_table_exists="replace")

但是，我得到了以下错误:

OperationalError: (sqlite3.OperationalError) unable to open database file

编辑:根据答案，使用pip install polars[sqlalchemy]命令安装SQLAlChemy.

推荐答案

使用SQLite时，斜杠的数量取决于您访问SQLite文件的方式.如果你按照 comments 中的建议使用两个斜杠，你会看到:

sqlalchemy.exc.ArgumentError: Invalid SQLite URL: sqlite://test.db
Valid SQLite URL forms are:
 sqlite:///:memory: (or, sqlite://)
 sqlite:///relative/path/to/file.db
 sqlite:////absolute/path/to/file.db

在您的例子中，您似乎希望使用3个斜杠，而不是4.请try 如下所示:

df.write_database("test_table", f"sqlite:///test.db", if_table_exists="replace")

这将是一个工作示例，基于您提供的示例代码片段:

import polars as pl
import sqlite3

conn = sqlite3.connect("test.db")
df = pl.DataFrame({"col1": [1, 2, 3]})
df.write_database("test_table", f"sqlite:///test.db", if_table_exists="replace")