import baostock as bs import pandas as pd import pymysql

登陆系统

lg = bs.login()

显示登陆返回信息

print('login respond error_code:' + lg.error_code) print('login respond error_msg:' + lg.error_msg)

连接数据库

conn = pymysql.connect(host='localhost', port=3306, user='root', password='123456', db='baostock', charset='utf8mb4')

获取游标

cursor = conn.cursor()

获取最近4个季度的数据

yjyg_list = [] for year in range(2021, 2022): for quarter in range(1, 5): rs_yjyg = bs.query_profit_data(code='sh.600000', year=year, quarter=quarter) while (rs_yjyg.error_code == '0') & rs_yjyg.next(): yjyg_list.append(rs_yjyg.get_row_data()) result_yjyg = pd.DataFrame(yjyg_list, columns=rs_yjyg.fields)

打印输出

print(result_yjyg)

结果集插入到数据库

for index, row in result_yjyg.iterrows(): sql = "INSERT INTO yjyg VALUES ('sh.600000', {}, {}, '{}', {}, {}, {}, {}, {}, {}, {}, {}, {}, {}, {})".format( int(row[1]), int(row[2]), row[3], float(row[4]), float(row[5]), float(row[6]), float(row[7]), float(row[8]), float(row[9]), float(row[10]), float(row[11]), float(row[12]), float(row[13]), float(row[14])) cursor.execute(sql) conn.commit()

关闭游标和连接

cursor.close() conn.close()

登出系统

bs.logout()