pandas 读写sql

 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
37
38
39

import mysql.connector
import pandas as pd
from sqlalchemy import create_engine


# 创建连接
mydb = mysql.connector.connect(user=用户名, password=密码)

# 建立游标
mycursor = mydb.cursor()

# 使用 data_test 数据库
mycursor.execute("USE data_test")

# --------------------------------------------------------------------------------------
# 读 数据库中的数据
data = pd.read_sql("SELECT file_version,station_id FROM wind_hobs_basic", con=mydb)

print(data)

# 关闭游标
mycursor.close()
# 关闭连接
mydb.close()

# --------------------------------------------------------------------------------------
# 向 数据库 写数据
engine = create_engine('mysql+mysqlconnector://root:abc@127.0.0.1:3306/data_test')

con = engine.connect()

# 创建一个 dataframe格式 数据
df_data = pd.DataFrame([[1.2, '54513', 116.47, 39.8069, 31.3, 'LC']], columns=['file_version', 'station_id', 'lon', 'lat', 'alt', 'radar_type'])

# 写入数据库，如果已存在该表，则追加写入数据，不加索引
df_data.to_sql(name="表名", con=con, if_exists='append', index=False)

con.close()