1. 数据科学与图表
  2. 连接到数据库

连接到数据库

您希望可视化的数据可能存储在数据库中。让我们使用SQLAlchemy快速将数据库内容提取到pandas Dataframe格式,以便我们可以在gradio中使用它。

首先安装 pip install sqlalchemy,然后我们来看一些例子。

SQLite

from sqlalchemy import create_engine
import pandas as pd

engine = create_engine('sqlite:///your_database.db')

with gr.Blocks() as demo:
    gr.LinePlot(pd.read_sql_query("SELECT time, price from flight_info;", engine), x="time", y="price")

让我们看一个更互动的图表,涉及修改你的SQL查询的过滤器:

from sqlalchemy import create_engine
import pandas as pd

engine = create_engine('sqlite:///your_database.db')

with gr.Blocks() as demo:
    origin = gr.Dropdown(["DFW", "DAL", "HOU"], value="DFW", label="Origin")

    gr.LinePlot(lambda origin: pd.read_sql_query(f"SELECT time, price from flight_info WHERE origin = {origin};", engine), inputs=origin, x="time", y="price")

Postgres、mySQL 和其他数据库

如果您使用的是不同的数据库格式,您只需更换引擎,例如。

engine = create_engine('postgresql://username:password@host:port/database_name')
engine = create_engine('mysql://username:password@host:port/database_name')
engine = create_engine('oracle://username:password@host:port/database_name')