【python doris】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
1
2

然后，您可以使用以下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()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

请根据您的实际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()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36