使用Matplotlib为Pandas数据框中的不同值在散点上添加不同的颜色

import pandas as pd
import matplotlib.pyplot as plt

data = {'x': [1, 2, 3, 4, 5],
        'y': [2, 4, 6, 8, 10],
        'category': ['A', 'B', 'A', 'B', 'A']}
df = pd.DataFrame(data)

color_map = {'A': 'red', 'B': 'blue'}

plt.scatter(df['x'], df['y'], c=df['category'].map(color_map))

plt.legend(handles=[plt.scatter([], [], c='red'), plt.scatter([], [], c='blue')],
           labels=['Category A', 'Category B'])
plt.xlabel('X')
plt.ylabel('Y')

plt.show()