[python] matplotlib 막대 차트에 값 레이블 추가

다른 질문에 대한이 답변 에서 언급 된 기능을 기반 으로 막대 차트에 레이블을 배치하는 데 매우 일반적으로 적용 가능한 솔루션을 찾았습니다.

안타깝게도 다른 솔루션은 레이블과 막대 사이의 간격이 막대의 절대 단위로 지정 되거나 막대 높이에 따라 조정 되기 때문에 많은 경우에서 작동하지 않습니다 . 전자는 좁은 범위의 값에 대해서만 작동하고 후자는 한 플롯 내에서 일관되지 않은 간격을 제공합니다. 둘 다 로그 축에서 잘 작동하지 않습니다.

내가 제안한 솔루션은 스케일 (즉, 작은 숫자와 큰 숫자)과는 독립적으로 작동 points하며 오프셋에 시각적 단위 를 사용하기 때문에 음수 값과 로그 스케일로 레이블을 올바르게 배치 합니다.

이러한 경우 레이블의 올바른 배치를 보여주기 위해 음수를 추가했습니다.

각 막대의 높이 값이 레이블로 사용됩니다. 다른 레이블은 Simon의 for rect, label in zip(rects, labels)snippet 과 함께 쉽게 사용할 수 있습니다 .

import numpy as np
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)


def add_value_labels(ax, spacing=5):
    """Add labels to the end of each bar in a bar chart.

    Arguments:
        ax (matplotlib.axes.Axes): The matplotlib object containing the axes
            of the plot to annotate.
        spacing (int): The distance between the labels and the bars.
    """

    # For each bar: Place a label
    for rect in ax.patches:
        # Get X and Y placement of label from rect.
        y_value = rect.get_height()
        x_value = rect.get_x() + rect.get_width() / 2

        # Number of points between bar and label. Change to your liking.
        space = spacing
        # Vertical alignment for positive values
        va = 'bottom'

        # If value of bar is negative: Place label below bar
        if y_value < 0:
            # Invert space to place label below
            space *= -1
            # Vertically align label at top
            va = 'top'

        # Use Y value as label and format number with one decimal place
        label = "{:.1f}".format(y_value)

        # Create annotation
        ax.annotate(
            label,                      # Use `label` as label
            (x_value, y_value),         # Place label at end of the bar
            xytext=(0, space),          # Vertically shift label by `space`
            textcoords="offset points", # Interpret `xytext` as offset in points
            ha='center',                # Horizontally center label
            va=va)                      # Vertically align label differently for
                                        # positive and negative values.


# Call the function above. All the magic happens there.
add_value_labels(ax)

plt.savefig("image.png")

편집 : barnhillec이 제안한대로 함수에서 관련 기능을 추출했습니다 .

그러면 다음과 같은 출력이 생성됩니다.

그리고 로그 스케일 (및 로그 스케일링을 보여주기 위해 입력 데이터에 대한 일부 조정)을 사용하면 결과는 다음과 같습니다.