Thêm nhãn giá trị trên biểu đồ thanh matplotlib

Tôi bị mắc kẹt vào một cái gì đó mà tôi cảm thấy tương đối dễ dàng. Đoạn mã tôi mang đến dưới đây là một mẫu dựa trên một dự án lớn hơn mà tôi đang thực hiện. Tôi thấy không có lý do gì để đăng tất cả các chi tiết, vì vậy hãy chấp nhận cấu trúc dữ liệu mà tôi mang theo.

Về cơ bản, tôi đang tạo một biểu đồ thanh và tôi chỉ có thể tìm ra cách thêm nhãn giá trị trên các thanh (ở giữa thanh hoặc ngay trên nó). Tôi đã xem xét các mẫu trên web nhưng không triển khai thành công trên mã của riêng tôi. Tôi tin rằng giải pháp là với 'văn bản' hoặc 'chú thích', nhưng tôi: a) không biết nên sử dụng cái nào (và nói chung là chưa tìm ra khi nào nên dùng cái nào). b) không thể nhìn thấy để hiển thị các nhãn giá trị. Đánh giá cao sự giúp đỡ của bạn, mã của tôi dưới đây. Cảm ơn trước!

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
pd.set_option('display.mpl_style', 'default') 
%matplotlib inline

# 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))
fig = freq_series.plot(kind='bar')
fig.set_title('Amount Frequency')
fig.set_xlabel('Amount ($)')
fig.set_ylabel('Frequency')
fig.set_xticklabels(x_labels)

Đầu tiên freq_series.plottrả về một trục không phải là một hình, vì vậy để làm cho câu trả lời của tôi rõ ràng hơn một chút, tôi đã thay đổi mã đã cho của bạn để tham chiếu đến nó axthay vì figphù hợp hơn với các ví dụ mã khác.

Bạn có thể lấy danh sách các thanh được sản xuất trong cốt truyện từ ax.patchesthành viên. Sau đó, bạn có thể sử dụng kỹ thuật được trình bày trong ví dụ thư viện nàymatplotlib để thêm các nhãn bằng ax.textphương pháp này.

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)

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')

Điều này tạo ra một âm mưu được gắn nhãn trông giống như:

Dựa trên một tính năng được đề cập trong câu trả lời này cho một câu hỏi khác, tôi đã tìm thấy một giải pháp áp dụng rất phổ biến để đặt nhãn trên biểu đồ thanh.

Rất tiếc, các giải pháp khác không hoạt động trong nhiều trường hợp, vì khoảng cách giữa nhãn và thanh được tính theo đơn vị tuyệt đối của các thanh hoặc được chia tỷ lệ theo chiều cao của thanh . Cái trước chỉ hoạt động cho một phạm vi giá trị hẹp và cái sau cho khoảng cách không nhất quán trong một ô. Cả hai đều không hoạt động tốt với các trục logarit.

Giải pháp mà tôi đề xuất hoạt động độc lập với tỷ lệ (tức là đối với số lượng nhỏ và lớn) và thậm chí đặt đúng nhãn cho các giá trị âm và với thang đo logarit vì nó sử dụng đơn vị trực quan pointscho các hiệu số.

Tôi đã thêm một số âm để giới thiệu vị trí chính xác của các nhãn trong trường hợp như vậy.

Giá trị chiều cao của mỗi thanh được sử dụng làm nhãn cho nó. Các nhãn khác có thể dễ dàng được sử dụng với đoạn mã của Simonfor rect, label in zip(rects, labels) .

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")

Chỉnh sửa: Tôi đã trích xuất chức năng có liên quan trong một chức năng, như được đề xuất bởibarhillec .

Điều này tạo ra kết quả sau:

Và với thang đo logarit (và một số điều chỉnh đối với dữ liệu đầu vào để hiển thị tỉ lệ logarit), đây là kết quả:

Dựa trên câu trả lời (tuyệt vời!) Ở trên, chúng tôi cũng có thể tạo biểu đồ thanh ngang chỉ với một vài điều chỉnh:

# Bring some raw data.
frequencies = [6, -16, 75, 160, 244, 260, 145, 73, 16, 4, 1]

freq_series = pd.Series(frequencies)

y_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='barh')
ax.set_title('Amount Frequency')
ax.set_xlabel('Frequency')
ax.set_ylabel('Amount ($)')
ax.set_yticklabels(y_labels)
ax.set_xlim(-40, 300) # expand xlim to make labels easier to read

rects = ax.patches

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

    # Number of points between bar and label. Change to your liking.
    space = 5
    # Vertical alignment for positive values
    ha = 'left'

    # If value of bar is negative: Place label left of bar
    if x_value < 0:
        # Invert space to place label to the left
        space *= -1
        # Horizontally align label at right
        ha = 'right'

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

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

plt.savefig("image.png")

Nếu bạn chỉ muốn gắn nhãn các điểm dữ liệu phía trên thanh, bạn có thể sử dụng plt.annotate ()

Mã của tôi:

import numpy as np
import matplotlib.pyplot as plt

n = [1,2,3,4,5,]
s = [i**2 for i in n]
line = plt.bar(n,s)
plt.xlabel('Number')
plt.ylabel("Square")

for i in range(len(s)):
    plt.annotate(str(s[i]), xy=(n[i],s[i]), ha='center', va='bottom')

plt.show()

Bằng cách chỉ định căn chỉnh theo chiều ngang và dọc của 'center'và 'bottom'tương ứng, người ta có thể nhận được các chú thích ở giữa.

Nếu bạn chỉ muốn thêm Datapoints phía trên các thanh, bạn có thể dễ dàng làm điều đó với:

 for i in range(len(frequencies)): # your number of bars
    plt.text(x = x_values[i]-0.25, #takes your x values as horizontal positioning argument 
    y = y_values[i]+1, #takes your y values as vertical positioning argument 
    s = data_labels[i], # the labels you want to add to the data
    size = 9) # font size of datalabels