Matplotlib: Automati c маркировка рядом на гистограмме - PullRequest
1 голос
/ 12 апреля 2020

На основе следующего примера из matplotlib я создал функцию, которая отображает два еженедельных временных ряда в виде столбчатой ​​диаграммы рядом друг с другом. https://matplotlib.org/3.1.1/gallery/lines_bars_and_markers/barchart.html#sphx -glr-gallery-lines-bars-and-markers-barchart-py

Моя проблема заключается в том, что я устанавливаю xticks явно, и это создает грязные xtick-метки. Есть ли способ заставить matplotlib явно выбирать xticks (положение и метки) в таком графике?

Я должен сказать, что я нахожу всю операцию с указанием позиции бара с помощью (x - width / 2) ) довольно нелегко получить параллельные панели - есть ли другие варианты (другие пакеты, кроме matplotlib или другие спецификации в matplotlib), чтобы избежать написания такого явного кода?

Ниже приведены код и результат. Я ищу решение, которое выбирает количество и размещение символов xticks и xticklabels, чтобы сделать его читаемым:

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


labels = ['W1-2020', 'W2-2020', 'W3-2020', 'W4-2020', 'W5-2020','W6-2020','W7-2020','W8-2020','W9-2020','W10-2020','W11-2020','W12-2020','W13-2020','W14-2020','W15-2020']
men_means = [20, 34, 30, 35, 27,18,23,29,27,29,38,28,17,28,23]
women_means = [25, 32, 34, 20, 25,27,18,23,29,27,29,38,19,20, 34]

x = np.arange(len(labels))  # the label locations
width = 0.35  # the width of the bars

fig, ax = plt.subplots()
rects1 = ax.bar(x - width/2, men_means, width, label='Men')
rects2 = ax.bar(x + width/2, women_means, width, label='Women')

# Add some text for labels, title and custom x-axis tick labels, etc.
ax.set_ylabel('Scores')
ax.set_title('Scores by group and gender')
ax.set_xticks(x)
ax.set_xticklabels(labels)
ax.legend()


def autolabel(rects):
    """Attach a text label above each bar in *rects*, displaying its height."""
    for rect in rects:
        height = rect.get_height()
        ax.annotate('{}'.format(height),
                    xy=(rect.get_x() + rect.get_width() / 2, height),
                    xytext=(0, 3),  # 3 points vertical offset
                    textcoords="offset points",
                    ha='center', va='bottom')


autolabel(rects1)
autolabel(rects2)

fig.tight_layout()

plt.show()

enter image description here

1 Ответ

1 голос
/ 12 апреля 2020

Решение 1: Использование Pandas

Вы можете сначала создать pandas DataFrame, а затем непосредственно построить несколько гистограмм. Форматирование надписей на оси x значительно более точное

df = pd.DataFrame(
    {'labels': labels,
     'men_means': men_means,
     'women_means': women_means
    })

df.plot(x="labels", y=["men_means", "women_means"], kind="bar")

enter image description here

Решение 2: Использование Seaborn (адаптировано из этот ответ)

import seaborn as sns

fig, ax = plt.subplots(figsize=(6, 4))
tidy = df.melt(id_vars='labels').rename(columns=str.title)
sns.barplot(x='Labels', y='Value', hue='Variable', data=tidy, ax=ax)
sns.despine(fig)
ax.tick_params(axis='x', labelrotation=90)

enter image description here

Чтобы скрыть только каждый n-й тик, вы можете сделать следующее в соответствии с этот ответ

n = 2
for label in ax.xaxis.get_ticklabels()[::n]:
    label.set_visible(False)

Чтобы показать каждый n-й ярлык, вы можете использовать следующий трюк

fig.canvas.draw()

n = 4
labels = [item.get_text() if i%n == 0 else "" for i, item in enumerate(ax.get_xticklabels())]

ax.set_xticklabels(labels);

enter image description here

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...