如何做列表或二维数组或矩阵的散点图?
问题描述
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()
相关文章