用python绘制散点图

今天下午学习了如何使用python绘制简单的散点图，写成博客分享一下。

在python中画散点图主要是用matplotlib模块中的scatter函数，先来看一下scatter函数的基本信息。

网址为：点击打开链接

可以看到scatter中有很多参数，经常使用的参数主要有以下几个：

c：

marker:

数据、代码和绘制的图如下。

数据（取第一列作为x，取第四列作为y）截图：

代码如下：

import matplotlib.pyplot as plt
import numpy as np

# 定义画散点图的函数
def draw_scatter(n, s):
	"""
	:param n: 点的数量，整数
	:param s:点的大小，整数
	:return: None
	"""
	# 加载数据
	data = np.loadtxt('results.txt', encoding='utf-8', delimiter=',')
	# 通过切片获取横坐标x1
	x1 = data[:, 0]
	# 通过切片获取纵坐标R
	y1 = data[:, 3]
	# 横坐标x2
	x2 = np.random.uniform(0, 5, n)
	# 纵坐标y2
	y2 = np.array([3] * n)
	# 创建画图窗口
	fig = plt.figure()
	# 将画图窗口分成1行1列，选择第一块区域作子图
	ax1 = fig.add_subplot(1, 1, 1)
	# 设置标题
	ax1.set_title('Result Analysis')
	# 设置横坐标名称
	ax1.set_xlabel('gamma-value')
	# 设置纵坐标名称
	ax1.set_ylabel('R-value')
	# 画散点图
	ax1.scatter(x1, y1, s=s, c='k', marker='.')
	# 画直线图
	ax1.plot(x2, y2, c='b', ls='--')
	# 调整横坐标的上下界
	plt.xlim(xmax=5, xmin=0)
	# 显示
	plt.show()

# 主模块
if __name__ == "__main__":
	# 运行
	draw_scatter(n=2000, s=20)

用python绘制散点图

欢迎大家关注DataLearner官方微信，接受最新的AI技术推送

相关博客

最热博客