PolarsでMySQLデータベースからデータを読み込む

Polarsではread_databaseメソッドを使ってMySQLなどのデータベースから読み込むことができます。

準備

特にMySQLデータベースから読み込むときにはconnectorxというエンジンを用いているため、これをインストールする必要があります。さらにpyarrowというものも必要でした：

$ pip install connectorx pyarrow

使い方

引数にはSQLクエリとuriを渡します。uriはMySQLの場合はmysql://{username}:{password}@{server}:{port}/{database}です。

import polars as pl

username = 'root'
password = ''
server = '127.0.0.1'
port = '3306'
database = 'test'

uri = f'mysql://{username}:{password}@{server}:{port}/{database}'
query = 'SELECT * FROM users'
df = pl.read_database(query, uri)