Matplotlib绘制带局部放大的多个子图

带局部放大的多个子图

# -*- coding: utf-8 -*-
"""
Created on Thu Oct 31 21:17:45 2019

@author: huang
"""

import xlrd
import numpy as np
import math
import matplotlib.pyplot as plt

from matplotlib.patches import ConnectionPatch
from matplotlib.font_manager import FontProperties
#from scipy import stats
#from pylab import * 
#mpl.rcParams['font.sans-serif'] = ['SimHei']

def Draw(Path1, Path2):
    # 定义字体
    fontdict1={'family':'Times New Roman',#宋体 常规
              'weight':'normal',
              'size':10,#7.5磅
          }
    
    # 数据
    dst_all,xlength = DistArr(Path1, Path2)
    # 绘制折线图
    # 参数
    Criterion = 3000
    Percent = 0.001
    ty1 = 15000
    startid = 900
    # 绘制画布
    plt.figure(num=1, figsize=(12, 5))
    
    # 激活第一个 subplot
    p1 = plt.subplot(1,  2,  1)
    x=range(0,xlength,1)
    y=dst_all[0:xlength]
    
    yMax = max(y)
    
    plt.plot(x,y)
    plt.xlabel('xxx',fontdict1)
    plt.ylabel('xxx',fontdict1)
    plt.axhline(Criterion,ls='--',color='black')
    plt.axis([0,xlength,0,yMax])
    
    # 激活第一个 subplot
    p2 = plt.subplot(1,  2,  2)
    
    x=range(startid,xlength,1)
    y=dst_all[startid:xlength]
    
    plt.plot(x,y)
    
    plt.xlabel('xxx',fontdict1)
    plt.ylabel('xxx',fontdict1)
    plt.axhline(Criterion,ls='--',color='black')
    #plt.axis([0,int(xlength*Percent),0,yMax])
    plt.axis([startid,xlength,0,yMax])
    
    
    # 画p1中即将被放大区域的方框
    tx0 = 0
    if Percent>0.05:
        tx1 = int(xlength*Percent) 
    else: 
        tx1 = int(xlength*0.06)
    ty0 = 0
    #ty1 = 5000
    
    sx = [tx0,tx1,tx1,tx0,tx0]
    sy = [ty0,ty0,ty1,ty1,ty0]
    p1.plot(sx,sy,"red")
    
    
    # 在两个子图之间画连接线
    xy2 = (10, yMax)
    xy1 = (tx1, ty1)
    con = ConnectionPatch(xyA=xy2,xyB=xy1,coordsA="data",coordsB="data",
    axesA=p2,axesB=p1)
    p1.add_artist(con)
    
    xy2 = (startid, 0)
    xy1 = (tx1, ty0)
    con = ConnectionPatch(xyA=xy2,xyB=xy1,coordsA="data",coordsB="data",
    axesA=p2,axesB=p1)
    p1.add_artist(con)

Path1 = 'xxx'
Path2 = 'xxx'

Draw(Path1, Path2)