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

【python doris】python连接doris数据库，并查询返回数据框

news 来源：原创 2024/9/20 12:13:00

连接Doris数据库时，由于Doris与MySQL协议兼容，因此可以使用与MySQL兼容的Python库。以下是一些常用的Python库，您可以使用它们来连接和操作Doris数据库：

pymysql是一个Python编写的MySQL客户端库，由于Doris与MySQL协议兼容，因此可以直接使用pymysql来连接Doris数据库。

在Python中，您可以使用pandas库将查询结果转换为数据框（DataFrame）。首先，确保您已经安装了pandas库。如果没有安装，可以使用pip进行安装：

pip install pandas
pip install pymysql

然后，您可以使用以下Python代码连接到Doris数据库，执行查询，并将结果转换为pandas DataFrame：

import pymysql
import pandas as pd
# Doris数据库连接参数
doris_host = 'your_doris_host'
doris_port = your_doris_port  # 注意：这里应该是整数
doris_user = 'your_username'
doris_password = 'your_password'
doris_db = 'your_database'
# 建立连接
conn = pymysql.connect(host=doris_host,port=doris_port,user=doris_user,passwd=doris_password,db=doris_db)
# 编写您的SQL查询
query = "SELECT * FROM your_table LIMIT 10;"
# 使用pandas的read_sql_query方法执行查询并转换为DataFrame
df = pd.read_sql_query(query, conn)
# 打印DataFrame
print(df)
# 关闭连接
conn.close()

请根据您的实际Doris数据库配置替换your_doris_host、your_doris_port、your_username、your_password、your_database和your_table。
在上面的代码中，我们使用了pandas的read_sql_query方法，它接受一个SQL查询和一个数据库连接对象，然后直接将查询结果转换为DataFrame。这样，您就可以方便地使用pandas提供的各种数据操作和分析功能了。

如果想用游标返回数据：

import pymysql# Doris数据库连接参数
doris_host = 'your_doris_host'
doris_port = your_doris_port  # 注意：这里应该是整数
doris_user = 'your_username'
doris_password = 'your_password'
doris_db = 'your_database'# 建立连接
conn = pymysql.connect(host=doris_host,port=doris_port,user=doris_user,passwd=doris_password,db=doris_db)# 创建游标对象
cursor = conn.cursor()# 编写您的SQL查询
query = "SELECT * FROM your_table LIMIT 10;"# 执行查询
cursor.execute(query)# 获取查询结果
rows = cursor.fetchall()# 打印查询结果
for row in rows:print(row)# 关闭游标和连接
cursor.close()
conn.close()