Saya sedang mengerjakan plot dengan dua sumbu yang mendukung pemilihan garis. Saya menggunakan matplotlib dan perintah twinx() . Sayangnya acara pick hanya dipanggil untuk artis di sumbu paling atas (lihat contoh di bawah).

import matplotlib.pyplot as plt
import numpy as np

def onPick(event):
    print(event.artist.get_label())
    
def pick():
    fig=plt.figure(figsize=(5, 4), dpi=100, tight_layout=True)
    axis_1=fig.add_subplot(111)
    axis_2=axis_1.twinx()

    axis_1.set_ylim(0, 10)
    axis_2.set_ylim(0, 10)

    x=np.array([1,2,3,4])
    y_1=np.array([1,1,1,1])
    y_2=y_1+4
    
    axis_1.plot(x, y_1, picker=5, label='line_1')
    axis_2.plot(x, y_2, picker=5, label='line_2')
    
    fig.canvas.mpl_connect('pick_event', onPick)
    plt.show()

if __name__=='__main__':
    pick()

Apakah ada cara untuk memilih garis dari sumbu di bawah ini?

1
EGuy 5 Juli 2020, 13:22

1 menjawab

Jawaban Terbaik

Mustahil. ^^ Saya menemukan solusinya. Saya tidak memilih dari kapak, saya memilih dari legenda.
Saya pikir ini adalah kompromi yang baik.

import matplotlib.pyplot as plt
import numpy as np
from numpy.random import rand


def onpick(event):
        print(event.artist.get_label())


if __name__ == '__main__':
    t=np.linspace(1, 10, 100)
    y1, y2=1*t, 2*t
    y3, y4=3*t, 4*t

    fig, ax1=plt.subplots()
    ax2=ax1.twinx()
    ax2._get_lines.prop_cycler = ax1._get_lines.prop_cycler # Send Color cycle state to second axis.
    
    line1, = ax1.plot(t, y1, lw=2, label='1 HZ')
    line2, = ax1.plot(t, y2, lw=2, label='2 HZ')
    line3, = ax2.plot(t, y3, lw=2, label='3 HZ')
    line4, = ax2.plot(t, y4, lw=2, label='4 HZ')
    
    leg=ax1.legend(handles=[line1, line2, line3, line4], bbox_to_anchor=(0,1.02,1,0.2), loc="lower left", mode="expand", borderaxespad=0, ncol=3)
    for line in leg.get_lines(): line.set_picker(5)
    
    fig.canvas.mpl_connect('pick_event', onpick)
    plt.show()
1
EGuy 13 Juli 2020, 15:25