import pandas as pd import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D plt.rcParams['font.sans-serif']=['SimHei'] iris = np.loadtxt(r"\iris.csv", delimiter=",", usecols=(0,1,2)) Setosa = iris[:50] Versicolor = iris[50:100] Virginica = iris[100:] fig = plt.figure() ax = fig.add_subplot(111, projection='3d') ax.scatter(Setosa[:,0], Setosa[:,1], Setosa[:,2], c='r', label='Setosa') ax.scatter(Versicolor[:,0], Versicolor[:,1], Versicolor[:,2], c='g', label='Versicolor') ax.scatter(Virginica[:,0], Virginica[:,1], Virginica[:,2], c='b', label='Virginica') ax.legend() ax.set_xlabel('花萼长度') ax.set_ylabel('花萼宽度') ax.set_zlabel('花瓣长度') ax.set_title('鸢尾花三维散点图') plt.show()