当前位置：首页 > news >正文

使用 Python 连接 PostgreSQL 数据库，从 `mimic - III` 数据库中筛选数据并导出特定的数据图表

news 来源：原创 2025/6/21 22:02:28

要使用 Python 连接 PostgreSQL 数据库，从 mimic - III 数据库中筛选数据并导出特定的数据图表，你可以按照以下步骤操作：

安装所需的库：psycopg2 用于连接 PostgreSQL 数据库，pandas 用于数据处理，matplotlib 用于绘制图表。
连接到 PostgreSQL 数据库。
执行 SQL 查询以筛选所需的数据。
使用 pandas 处理数据。
使用 matplotlib 绘制图表。
保存图表。

以下是一个示例代码：

import psycopg2
import pandas as pd
import matplotlib.pyplot as plt


def connect_to_database():
    try:
        # 建立数据库连接
        connection = psycopg2.connect(
            database="your_database",
            user="your_user",
            password="your_password",
            host="your_host",
            port="your_port"
        )
        return connection
    except (Exception, psycopg2.Error) as error:
        print("Error while connecting to PostgreSQL", error)
        return None


def fetch_data(connection):
    try:
        # 执行 SQL 查询
        query = "SELECT * FROM your_table LIMIT 100"  # 替换为你的实际查询
        df = pd.read_sql(query, connection)
        return df
    except (Exception, psycopg2.Error) as error:
        print("Error while fetching data", error)
        return None


def plot_and_save_data(df):
    if df is not None:
        # 假设绘制一个简单的柱状图，这里你可以根据需求修改
        df.plot(kind='bar')
        plt.title('Sample Data Chart')
        plt.xlabel('X-axis')
        plt.ylabel('Y-axis')
        plt.savefig('data_chart.png')
        print("Chart saved as data_chart.png")


if __name__ == "__main__":
    connection = connect_to_database()
    if connection:
        data = fetch_data(connection)
        plot_and_save_data(data)
        connection.close()

代码说明：

connect_to_database 函数：用于建立与 PostgreSQL 数据库的连接。你需要将 your_database、your_user、your_password、your_host 和 your_port 替换为实际的数据库信息。
fetch_data 函数：执行 SQL 查询并将结果存储在 pandas 的 DataFrame 中。你需要将 your_table 替换为实际的表名，并根据需求修改查询语句。
plot_and_save_data 函数：使用 matplotlib 绘制图表并保存为 data_chart.png。你可以根据需求修改图表类型和样式。
主程序：调用上述函数完成数据库连接、数据获取、图表绘制和保存操作，并在完成后关闭数据库连接。