本节知识重点
- 将连接数据库变为函数,便于直接写SQL即可调用
- 处理完后的数据表保存
将连接数据库变为函数
定义函数
def reader(query,db='data'):#query是sql语句,db是数据库名称,默认值是data
sql=query
engine= create_engine('mysql+pymysql://root:@localhost:3306/{0}?charset=utf8'.format(db))#.format()格式化
df=pd.read_sql(sql,engine)
return df
调用函数 ```python
将SQL数据库的值添加到df中
df_action=reader(‘select * from house_action’)
显示数据库中有哪些表
reader(‘show tables’)
Tables_in_data 0 house_action 1 member_order 2 operator
- 数据处理
```python
#数据处理
result=df_action.groupby(['operator_id','house_id']).count()[['action_type','thirdparty_types']]
- 数据保存
#数据保存 path='/Users/Desktop/result.csv' result.to_csv(path)