我已经编写了以下代码来生成两个不同的数据集。我想绘制X和y的图形,但不知道如何绘制。
import numpy as npimport matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_splitfrom sklearn.svm import SVCstd = [[0.5, 0], [0, 0.5]]X1 = np.vstack((np.random.multivariate_normal([2, -2], std, size=200),np.random.multivariate_normal([-2, 2], std, size=200)))y1 = np.zeros(X1.shape[0])X2 = np.vstack((np.random.multivariate_normal([2, 2], std, size=200),np.random.multivariate_normal([-2, -2], std, size=200)))y2 = np.ones(X2.shape[0])X = np.vstack((X1, X2))y = np.hstack((y1, y2))
我尝试了;
plt.scatter(X, y) plt.show
但正如你所猜测的,它们给我带来了错误。任何帮助将不胜感激。
回答:
以下两行代码将创建一个散点图来可视化你的数据
plt.scatter(X[:, 0], X[:,1], c=y)plt.show()