用pandas读取MySQL数据库中,某只股票100天的日交易数据,将查询结果返回的dataframe里的数据,直接保存到Excel文件中
时间: 2024-03-21 16:41:04 浏览: 52
Pandas读取MySQL数据到DataFrame的方法
好的,我知道了。下面是实现的步骤:
1. 首先需要安装pandas和mysql-connector-python这两个库,可以使用pip进行安装:
```
pip install pandas
pip install mysql-connector-python
```
2. 然后连接MySQL数据库,读取数据,生成DataFrame:
```
import mysql.connector
import pandas as pd
# 连接数据库
cnx = mysql.connector.connect(user='username', password='password',
host='localhost',
database='database_name')
# 读取数据到DataFrame
query = "SELECT * FROM stock_data WHERE stock_code='000001' LIMIT 100"
df = pd.read_sql(query, con=cnx)
# 关闭连接
cnx.close()
```
上面的代码中,`user`、`password`、`host`、`database`需要替换为自己的数据库连接信息。`stock_data`是存储股票交易数据的表名,`stock_code='000001'`表示查询股票代码为000001的数据,`LIMIT 100`表示限制查询结果为100条。
3. 将DataFrame里的数据保存到Excel文件中:
```
# 保存到Excel文件中
df.to_excel('stock_data.xlsx', index=False)
```
上面的代码中,`stock_data.xlsx`是保存的Excel文件名,`index=False`表示不保存行索引。
完整代码如下:
```
import mysql.connector
import pandas as pd
# 连接数据库
cnx = mysql.connector.connect(user='username', password='password',
host='localhost',
database='database_name')
# 读取数据到DataFrame
query = "SELECT * FROM stock_data WHERE stock_code='000001' LIMIT 100"
df = pd.read_sql(query, con=cnx)
# 关闭连接
cnx.close()
# 保存到Excel文件中
df.to_excel('stock_data.xlsx', index=False)
```
阅读全文