如何做列表或二维数组或矩阵的散点图?

2022-05-24 00:00:00 python matplotlib scatter-plot

问题描述

 X=[[0,3,4,0,1,1],
    [0,0,0,5,1,1],
    [6,7,0,8,1,1],
    [3,6,1,5,6,1]]

Y=[12,15,11,10]

我有如何绘制散点图以可视化X、Y以检查该关系的列表??


解决方案

您的X方向错误,我正在使用NumPy数组轻松地对其进行转置。然后,只需将其绘制在同一轴上,更改颜色,这样您就可以看到什么是什么。

import numpy as np
import matplotlib.pyplot as plt

x_arr = np.array(X)
y = np.array(Y)

fig, ax = plt.subplots()

colors=list('bgrcmykw')

for i, x in enumerate(x_arr.T):
    ax.scatter(x,y, c=colors[i])

plt.show()

相关文章