1、绘制不同颜色的点(二维)
import matplotlib.pyplot as pltif __name__ == '__main__':# 准备数据x = [1, 2, 3, 4, 5] # X轴上的点y = [1, 4, 9, 16, 25] # Y轴上的点,这里以x的平方为例colors = ['red', 'green', 'blue', 'yellow', 'purple'] # 点的颜色列表# 绘制点for i in range(len(x)):plt.scatter(x[i], y[i], color=colors[i]) # 使用scatter函数绘制点,并指定颜色# 添加标题和坐标轴标签plt.title('myTitle')plt.xlabel('X axis')plt.ylabel('Y axis')# 显示图形plt.show()
效果:
还可以绘制多个画布:
import matplotlib.pyplot as pltif __name__ == '__main__':# 准备数据x = [1, 2, 3, 4, 5] # X轴上的点y = [1, 4, 9, 16, 25] # Y轴上的点,这里以x的平方为例colors = ['red', 'green', 'blue', 'yellow', 'purple'] # 点的颜色列表fig = plt.figure() # 创建一个图形画布,可以增加子图形ax1 = fig.add_subplot(1, 2, 1) # 1行2列,第1个位置ax1.set_title('Subplot 1')ax1.scatter(x,y, s=7, c='red') # 绘制点;s表示点的大小ax2 = fig.add_subplot(1, 2, 2) # 1行2列,第2个位置ax2.set_title('Subplot 2')ax2.scatter(x, y, s=1, c='green') # 绘制点;s表示点的大小# 显示图形plt.show()