Python连接Doris数据库,查询并返回数据框【教程】
连接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()
作者:东华果汁哥