matplotlib雙y周實例e

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import pymysql
import warnings
import random

warnings.filterwarnings("ignore")


def read_db(path):
close = pd.read_csv(path)
info = close[:-6]
zhi_list = []
for x in info['qx_zu'][20:]:
x_index = info[info['qx_zu'] == x].index
all_shuju = info[x_index[0] - 20:x_index[0] - 1]
zhi = np.mean(all_shuju['qx_zu']) - 2 * (np.std(all_shuju['qx_zu']))
info[21:]['cha'] = zhi
zhi_list.append(zhi)
info['cha'] = 0
info['cha'][20:] = np.array(zhi_list)

info['jieguo'] = info['qx_zu'] < info['cha']
info['jieguo'][:20] = 0
x_true_index = info[info['jieguo'] == True].index
x_false_index = info[info['jieguo'] == False].index
x_small = info.loc[x_true_index]
x_small['jieguo'] = x_small['close']
# x_small['jieguo'] = 1
x_big = info.loc[x_false_index]
x_big['jieguo'] = 0
info_all = x_big.append(x_small)
info_all = info_all.sort_index()
# print(list(info_all['jieguo']))
# info_all.to_csv('info_all.csv')
info_all['date'] = pd.to_datetime(info_all['date'], format='%Y/%m/%d')
print(info_all['date'])
return info_all


def show_info(info):
"""
畫圖展示
:param info:
:return:
"""

plt.rcParams['font.sans-serif'] = ['SimHei'] # 設置全局顯示漢子
plt.tick_params(axis='x', labelsize=10) # 設置x軸標籤大小
plt.plot(info['date'], info['close'], 'black')
plt.legend(loc=4)

plt.xlabel('時間') # 設置x軸標題

plt.ylabel('滬深300收盤價', color='black') # 設置Y1軸標題
# plt.xticks(rotation=110) # 設置時間標籤顯示格式
plt.twinx() # 添加一條Y軸，
plt.axis('off') #不顯示刻度
plt.ylabel('預警信號', color='r') # 設置Y2軸標題

plt.bar(info['date'], info['jieguo'], label='預警信號', color='r')

plt.legend(loc='upper right') # 右上
# 保存圖片
# plt.savefig('image/{}.png'.format('滬深300收盤價'))
plt.show()


if __name__ == '__main__':
show_info(info=read_db('h300_close.csv'))