Стрелки между участками matplotlib - PullRequest
Новая
       2

Стрелки между участками matplotlib

1 голос
/ 23 марта 2020

Я решил немного поиграться с этим примером кода. Мне удалось выяснить, как нарисовать прямую линию между двумя вспомогательными участками, даже если линия находится за пределами одного из вспомогательных участков. 

import matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np

fig = plt.figure(figsize=(10, 5))
ax1 = fig.add_subplot(121)
ax2 = fig.add_subplot(122)
axs = [ax1, ax2]

# Fixing random state for reproducibility
np.random.seed(19680801)

# generate some random test data
all_data = [np.random.normal(0, std, 100) for std in range(6, 10)]

# plot violin plot
axs[0].violinplot(all_data,
                  showmeans=False,
                  showmedians=True)
axs[0].set_title('Violin plot')

# plot box plot
axs[1].boxplot(all_data)
axs[1].set_title('Box plot')

# adding horizontal grid lines
for ax in axs:
    ax.yaxis.grid(True)
    ax.set_xticks([y + 1 for y in range(len(all_data))])
    ax.set_xlabel('Four separate samples')
    ax.set_ylabel('Observed values')

for tick in ax.xaxis.get_major_ticks():
    tick.label.set_fontsize(20)
plt.setp(axs[0], xticklabels=['x1', 'x2', 'x3', 'x4'])

transFigure = fig.transFigure.inverted()
coord1 = transFigure.transform(ax1.transData.transform([5,10]))
coord2 = transFigure.transform(ax2.transData.transform([2,-10]))
line = mpl.lines.Line2D((coord1[0],coord2[0]),(coord1[1],coord2[1]),
                        c='k', lw=5, transform=fig.transFigure)
fig.lines.append(line)

Да, добавленная линия уродлива, но я просто хотел чтобы заставить его функционировать.

Однако, что я действительно хотел бы сделать, так это сделать стрелку между подсюжетами, и я не могу понять, как без фальсификации жюри мои собственные хвосты стрел. Есть ли способ сделать это, используя класс matplotlib.pyplot.arrow?

...