自定義分類圖例matplotlib-有解無憂

我制作了我的資料集的模型版本來說明我的問題。

我有一個這樣的圖表：

import pandas as pd
import matplotlib.pyplot as plt

data = {'x': [0, 1, 2, 3],
        'y': [3, 2, 1, 0],
        'cat1': ['A', 'B', 'A', 'B'],
        'cat2': ['f', 'g', 'h', 'i']}

df = pd.DataFrame(data)

colors = {'A':'tab:red',
          'B':'tab:blue'}

markers = {'f':"o",
           'g':"v",
           'h':"s",
           'i':" "}

fig, ax = plt.subplots()
for i in range(df.shape[0]):
  ax.scatter(df.loc[i,'x'], 
             df.loc[i,'y'], 
             color=colors[df.loc[i,'cat1']], 
             marker=markers[df.loc[i,'cat2']],
             label = df.loc[i, 'cat2'])
  
ax.legend()

自定義分類圖例 matplotlib

但我正在尋找這樣的傳奇：

自定義分類圖例 matplotlib

誰能給我一些關于如何解決這個問題的提示？此外，如果最終情節中的圖例在右側的情節外的框中會更好。

uj5u.com熱心網友回復：

要添加額外的線（A，B），重新排列順序并將圖例移到圖表外，這些是您在繪圖后可以遵循的步驟。

使用為新條目和現有條目創建自定義圖例條目Line2D
繪制圖例并使用bbox_to_anchor將圖例移動到您需要的位置。如果要移動位置，可以調整 bbox 內的坐標
調整 A、B 的標簽，因為它們只是文本（無標記），以便水平對齊到中間（set_position()如果您想進一步向左/向右移動，可以調整

代碼

import pandas as pd
import matplotlib.pyplot as plt

data = {'x': [0, 1, 2, 3],
        'y': [3, 2, 1, 0],
        'cat1': ['A', 'B', 'A', 'B'],
        'cat2': ['f', 'g', 'h', 'i']}

df = pd.DataFrame(data)

colors = {'A':'tab:red',
          'B':'tab:blue'}

markers = {'f':"o",
           'h':"s",
           'g':"v",
           'i':" "}

fig, ax = plt.subplots()
for i in range(df.shape[0]):
    ax.scatter(df.loc[i,'x'], df.loc[i,'y'], 
        color=colors[df.loc[i,'cat1']], 
        marker=markers[df.loc[i,'cat2']],
        label = df.loc[i, 'cat2']

## Create legend handle entries for each of the items
from matplotlib.lines import Line2D
title = Line2D([0], [0], linestyle="none", marker="")
f = Line2D([0], [0], linestyle="none", marker="o", markersize=10, markeredgecolor='tab:red', markerfacecolor="tab:red")
g = Line2D([0], [0], linestyle="none", marker="v", markersize=10,  markeredgecolor='tab:blue', markerfacecolor="tab:blue")
h = Line2D([0], [0], linestyle="none", marker="s", markersize=10,  markeredgecolor='tab:red', markerfacecolor="tab:red")
i = Line2D([0], [0], linestyle="none", marker=" ", markersize=10,  markeredgecolor='tab:blue', markerfacecolor="tab:blue")

## Plot in order you want, bbox to set legend box outside
leg=ax.legend((title, f, h, title, g, i), ('A', 'f', 'h', 'B', 'g','i'), bbox_to_anchor=(1.16, 1.03))

## Adjust position of A and B so that they are in middle
for item, label in zip(leg.legendHandles, leg.texts):
    if label._text in ['A', 'B']:
        width=item.get_window_extent(fig.canvas.get_renderer()).width
        label.set_ha('left')
        label.set_position((-width/2,0)) ## Adjust here to move left/right

陰謀

自定義分類圖例 matplotlib

轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/512072.html

標籤：Python熊猫matplotlib

上一篇：使用matplot在圖表欄中添加點

下一篇：Python：注釋條形圖的最大值：字串x，inty