كيفية إضافة وسيلة إيضاح إلى scatterplot في matplotlib

يمكنك استخدام بناء الجملة التالي لإضافة وسيلة إيضاح إلى مخطط التشتت في Matplotlib:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend
plt. legend ( * scatter.legend_elements ())

توضح الأمثلة التالية كيفية استخدام بناء الجملة هذا عمليًا.

مثال 1: وسيلة إيضاح Scatterplot مع القيم

يوضح المثال التالي كيفية إنشاء مخطط مبعثر حيث تعرض وسيلة الإيضاح القيم:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend with values
plt. legend ( * scatter.legend_elements ()) 

مثال 2: وسيلة إيضاح Scatterplot مع الفئات

يوضح المثال التالي كيفية إنشاء مخطط مبعثر حيث تعرض وسيلة الإيضاح أسماء الفئات:

 import matplotlib. pyplot as plt
from matplotlib. colors import ListedColormap

#define data
x = [3, 4, 4, 6, 8, 9]
y = [12, 14, 17, 16, 11, 13]

#define values, classes, and colors to map 
values = [0, 0, 1, 2, 2, 2]
classes = [' A ', ' B ', ' C ']
colors = ListedColormap([' red ', ' blue ', ' purple '])

#create scatterplot
scatter = plt. scatter (x, y, c=values, cmap=colors)

#add legend with class names
plt. legend (handles=scatter. legend_elements ()[ 0 ], labels=classes)

لاحظ أن هذه وسيلة الإيضاح تعرض أسماء الفئات التي حددناها (A، B، C) بدلاً من القيم (0، 1، 2) التي حددناها.

مصادر إضافية

كيفية زيادة حجم قطعة الأرض في Matplotlib
كيفية ضبط موضع العنوان في Matplotlib
كيفية تعيين نطاقات المحور في Matplotlib