Как сделать так, чтобы метки появлялись только над определенными полосами - PullRequest
Новая
       31

Как сделать так, чтобы метки появлялись только над определенными полосами

1 голос
/ 14 апреля 2019

Я хотел бы добавить собственную метку над определенными столбцами на этом графике.Как добавить метки только к полосе 4 и полосе 8 в этом примере: 

import pandas as pd
import matplotlib.pyplot as plt

# Bring some raw data.
frequencies = [6, 16, 75, 160, 244, 260, 145, 73, 16, 4, 1]
# In my original code I create a series and run on that, 
# so for consistency I create a series from the list.
freq_series = pd.Series.from_array(frequencies)

x_labels = [108300.0, 110540.0, 112780.0, 115020.0, 117260.0, 119500.0,
            121740.0, 123980.0, 126220.0, 128460.0, 130700.0]

# Plot the figure.
plt.figure(figsize=(12, 8))
ax = freq_series.plot(kind='bar')
ax.set_title('Amount Frequency')
ax.set_xlabel('Amount ($)')
ax.set_ylabel('Frequency')
ax.set_xticklabels(x_labels)

rects = ax.patches

# Make some labels.
labels = ["label%d" % i for i in xrange(len(rects))]

for rect, label in zip(rects, labels):
    height = rect.get_height()
    ax.text(rect.get_x() + rect.get_width() / 2, height + 5, label,
            ha='center', va='bottom')

С этим кодом я получаю метку над каждой полосой.Но мне нужны только надписи (например, со значениями 34 и 55) над четвертым и восьмым столбцом.

Как лучше всего это сделать?

Вот где я нашел этот пример: Добавление меток значений на гистограмму matplotlib

1 Ответ

3 голосов
/ 15 апреля 2019

Этот код был написан для Python 2 и намного более ранней версии pandas;Я немного исправил это для работы с Python 3.

Есть и другие способы сделать это, но не слишком сильно меняя структуру кода, вы можете указать индексы столбчатых меток, которые вы используете.хочу построить график, показанный ниже как which, и нанести их на график, только если они соответствуют: 

import pandas as pd
import matplotlib.pyplot as plt

# Bring some raw data.
frequencies = [6, 16, 75, 160, 244, 260, 145, 73, 16, 4, 1]
# In my original code I create a series and run on that, 
# so for consistency I create a series from the list.
freq_series = pd.Series(frequencies)

x_labels = [108300.0, 110540.0, 112780.0, 115020.0, 117260.0, 119500.0,
            121740.0, 123980.0, 126220.0, 128460.0, 130700.0]

# Plot the figure.
fig, ax = plt.subplots(figsize=(12, 8))
freq_series.plot(kind='bar', ax=ax)
ax.set_title('Amount Frequency')
ax.set_xlabel('Amount ($)')
ax.set_ylabel('Frequency')
ax.set_xticklabels(x_labels)

rects = ax.patches

# Make some labels.
labels = [f'label{i}' for i in range(len(rects))]

which = [3, 7]

for index, rect in enumerate(rects):
    if index in which:
        height = rect.get_height()
        ax.text(rect.get_x() + rect.get_width() / 2, height + 5, height,
                ha='center', va='bottom')

enter image description here

...