温馨提示×

Matplotlib怎么将数据点按大小或颜色映射到其他变量

小亿
83
2024-05-21 11:29:29
栏目: 编程语言

在Matplotlib中,可以使用scatter方法来绘制散点图,并通过传入参数cs来将数据点按颜色或大小映射到其他变量。以下是一个示例代码:

import matplotlib.pyplot as plt
import numpy as np

# 生成随机数据
np.random.seed(0)
x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)
sizes = 1000 * np.random.rand(100)

plt.scatter(x, y, c=colors, s=sizes, alpha=0.5, cmap='viridis')
plt.colorbar()  # 添加颜色条

plt.show()

在上面的示例代码中,c=colors将数据点按颜色映射到colors变量的值,s=sizes将数据点按大小映射到sizes变量的值。可以通过调整sizescolors的值来改变数据点的大小和颜色。alpha参数用于设置数据点的透明度,cmap参数用于设置颜色映射的颜色映射方案。最后,使用colorbar方法添加颜色条,显示颜色映射的范围。

0