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

07-19 1175阅读

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

【python doris】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()
VPS购买请点击我

文章版权声明:除非注明,否则均为主机测评原创文章,转载或复制请以超链接形式并注明出处。

目录[+]