Как записать в базу данных SQLite, используя Polars в Python?

Вот пример записи/чтения таблиц sqlite с использованием поляров.

Записать таблицу в базу данных (необходимо установить sqlalchemy).

import polars as pl

df = pl.DataFrame({"a": [1, 2, 3]})

df.write_database(
    "my_table",
    connection = "sqlite:///database.db",
    if_table_exists = "replace"
)

Чтение таблицы из базы данных (необходимо установить Connectorx).

pl.read_database_uri(query = "SELECT * FROM my_table", uri = "sqlite://database.db")

Обратите внимание на формат URI подключения.

• sqlite:///:memory: (или sqlite://)
• sqlite:///relative/path/to/file.db
• sqlite:////absolute/path/to/file.db
• sqlite:///c:/absolute/path/on/windows/file.db (благодарность @Thomas)

При работе с SQLite количество косых черт зависит от того, как вы получаете доступ к файлу sqlite. Если вы используете две косые черты, как предложено в комментариях, вы увидите следующее:

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. Вместо этого попробуйте следующее:

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")