pandas.read_sql_query() 函数执行 SQL 查询并将结果集存为集合
时间: 2024-02-09 16:13:10 浏览: 322
`pandas.read_sql_query()` 函数执行 SQL 查询并将结果集存储为 Pandas DataFrame 而不是集合。DataFrame 是 Pandas 中最常用的数据结构之一,它类似于电子表格或 SQL 表格,可以方便地进行数据清洗、转换和分析。
如果你想将 DataFrame 中的数据转换为 Python 集合,可以使用 DataFrame 的 `values` 属性,它会返回一个包含 DataFrame 中所有行的二维 numpy 数组。你可以将这个数组转换为 Python 列表或集合,例如:
```python
import pandas as pd
import sqlite3
# 连接到数据库
conn = sqlite3.connect('example.db')
# 执行 SQL 查询并将结果读取到 DataFrame 中
df = pd.read_sql_query("SELECT * from my_table", conn)
# 将 DataFrame 中的数据转换为 Python 集合
data = set([tuple(row) for row in df.values])
# 关闭数据库连接
conn.close()
```
这将执行一个名为 "my_table" 的表的查询,并将结果读取到一个名为 `df` 的 Pandas DataFrame 中。然后,我们使用 DataFrame 的 `values` 属性将数据转换为一个二维 numpy 数组,并使用列表推导式将每行转换为一个元组,并将它们添加到一个集合中。现在,`data` 变量包含了一个不可变的集合,其中每个元素都是一个包含一行数据的元组。
阅读全文