Chú thích chồng chéo Matplotlib

Question

Tôi muốn chú thích các thanh trong biểu đồ bằng một số văn bản nhưng nếu các thanh nằm gần nhau và có chiều cao tương đương, các chú thích ở trên ea. khác và do đó khó đọc (các tọa độ cho các chú thích được lấy từ vị trí và chiều cao của thanh).

Có cách nào để thay đổi một trong số chúng nếu có va chạm không?

Edit: Các thanh rất mỏng và rất gần đôi khi vì vậy chỉ cần sắp xếp theo chiều dọc không giải quyết được vấn đề ...

Một bức tranh có thể làm sáng tỏ điều:

Answer 1

Tôi đã viết giải pháp nhanh, kiểm tra từng vị trí chú thích đối với các hộp giới hạn mặc định cho tất cả các chú thích khác. Nếu có va chạm, nó sẽ thay đổi vị trí của nó sang vị trí xung đột có sẵn miễn phí tiếp theo. Nó cũng đặt trong mũi tên đẹp.

Đối với một ví dụ khá cực đoan, nó sẽ sản xuất này (không ai trong số những con số trùng nhau):

Thay vì điều này:

Đây là mã:

import numpy as np 
import matplotlib.pyplot as plt 
from numpy.random import * 

def get_text_positions(x_data, y_data, txt_width, txt_height): 
    a = zip(y_data, x_data) 
    text_positions = y_data.copy() 
    for index, (y, x) in enumerate(a): 
     local_text_positions = [i for i in a if i[0] > (y - txt_height) 
          and (abs(i[1] - x) < txt_width * 2) and i != (y,x)] 
     if local_text_positions: 
      sorted_ltp = sorted(local_text_positions) 
      if abs(sorted_ltp[0][0] - y) < txt_height: #True == collision 
       differ = np.diff(sorted_ltp, axis=0) 
       a[index] = (sorted_ltp[-1][0] + txt_height, a[index][1]) 
       text_positions[index] = sorted_ltp[-1][0] + txt_height 
       for k, (j, m) in enumerate(differ): 
        #j is the vertical distance between words 
        if j > txt_height * 2: #if True then room to fit a word in 
         a[index] = (sorted_ltp[k][0] + txt_height, a[index][1]) 
         text_positions[index] = sorted_ltp[k][0] + txt_height 
         break 
    return text_positions 

def text_plotter(x_data, y_data, text_positions, axis,txt_width,txt_height): 
    for x,y,t in zip(x_data, y_data, text_positions): 
     axis.text(x - txt_width, 1.01*t, '%d'%int(y),rotation=0, color='blue') 
     if y != t: 
      axis.arrow(x, t,0,y-t, color='red',alpha=0.3, width=txt_width*0.1, 
         head_width=txt_width, head_length=txt_height*0.5, 
         zorder=0,length_includes_head=True) 

Dưới đây là mã sản xuất các ô này, cho biết cách sử dụng:

#random test data: 
x_data = random_sample(100) 
y_data = random_integers(10,50,(100)) 

#GOOD PLOT: 
fig2 = plt.figure() 
ax2 = fig2.add_subplot(111) 
ax2.bar(x_data, y_data,width=0.00001) 
#set the bbox for the text. Increase txt_width for wider text. 
txt_height = 0.04*(plt.ylim()[1] - plt.ylim()[0]) 
txt_width = 0.02*(plt.xlim()[1] - plt.xlim()[0]) 
#Get the corrected text positions, then write the text. 
text_positions = get_text_positions(x_data, y_data, txt_width, txt_height) 
text_plotter(x_data, y_data, text_positions, ax2, txt_width, txt_height) 

plt.ylim(0,max(text_positions)+2*txt_height) 
plt.xlim(-0.1,1.1) 

#BAD PLOT: 
fig = plt.figure() 
ax = fig.add_subplot(111) 
ax.bar(x_data, y_data, width=0.0001) 
#write the text: 
for x,y in zip(x_data, y_data): 
    ax.text(x - txt_width, 1.01*y, '%d'%int(y),rotation=0) 
plt.ylim(0,max(text_positions)+2*txt_height) 
plt.xlim(-0.1,1.1) 

plt.show() 

Answer 2

Một lựa chọn là xoay văn bản/chú thích, được đặt bởi từ khóa/thuộc tính rotation. Trong ví dụ sau, tôi xoay văn bản 90 độ để đảm bảo rằng nó sẽ không va chạm với văn bản lân cận. Tôi cũng đặt va (viết tắt của verticalalignment) từ khóa, do đó văn bản được trình bày ở trên quầy bar (trên điểm mà tôi sử dụng để xác định văn bản):

import matplotlib.pyplot as plt 

data = [10, 8, 8, 5] 

fig = plt.figure() 
ax = fig.add_subplot(111) 
ax.bar(range(4),data) 
ax.set_ylim(0,12) 
# extra .4 is because it's half the default width (.8): 
ax.text(1.4,8,"2nd bar",rotation=90,va='bottom') 
ax.text(2.4,8,"3nd bar",rotation=90,va='bottom') 

plt.show() 

Kết quả là con số sau đây:

Xác định theo chương trình nếu có sự va chạm giữa các chú thích khác nhau là một quá trình phức tạp hơn. Đây có thể là một câu hỏi riêng biệt: Matplotlib text dimensions.

Answer 3

Tùy chọn khác sử dụng thư viện của tôi adjustText, được viết riêng cho mục đích này (https://github.com/Phlya/adjustText). Tôi nghĩ rằng nó có thể chậm hơn đáng kể mà câu trả lời được chấp nhận (nó làm chậm đáng kể với rất nhiều thanh), nhưng nhiều hơn nữa chung chung và cấu hình.

from adjustText import adjust_text 
np.random.seed(2017) 
x_data = np.random.random_sample(100) 
y_data = np.random.random_integers(10,50,(100)) 

f, ax = plt.subplots(dpi=300) 
bars = ax.bar(x_data, y_data, width=0.001, facecolor='k') 
texts = [] 
for x, y in zip(x_data, y_data): 
    texts.append(plt.text(x, y, y, horizontalalignment='center', color='b')) 
adjust_text(texts, add_objects=bars, autoalign='y', expand_objects=(0.1, 1), 
      only_move={'points':'', 'text':'y', 'objects':'y'}, force_text=0.75, force_objects=0.1, 
      arrowprops=dict(arrowstyle="simple, head_width=0.25, tail_width=0.05", color='r', lw=0.5, alpha=0.5)) 
plt.show() 

Nếu chúng ta cho phép autoalignment cùng trục x, nó được thậm chí tốt hơn (tôi chỉ cần để giải quyết một vấn đề nhỏ mà nó không thích đặt nhãn trên các điểm và không một chút để các bên...).

np.random.seed(2017) 
x_data = np.random.random_sample(100) 
y_data = np.random.random_integers(10,50,(100)) 

f, ax = plt.subplots(dpi=300) 
bars = ax.bar(x_data, y_data, width=0.001, facecolor='k') 
texts = [] 
for x, y in zip(x_data, y_data): 
    texts.append(plt.text(x, y, y, horizontalalignment='center', size=7, color='b')) 
adjust_text(texts, add_objects=bars, autoalign='xy', expand_objects=(0.1, 1), 
      only_move={'points':'', 'text':'y', 'objects':'y'}, force_text=0.75, force_objects=0.1, 
      arrowprops=dict(arrowstyle="simple, head_width=0.25, tail_width=0.05", color='r', lw=0.5, alpha=0.5)) 
plt.show() 

(tôi đã phải điều chỉnh một số thông số ở đây, tất nhiên)