python画散点图

1. 二维散点图

import matplotlib.pyplot as plt
import numpy as np
data=np.random.randint(0,100,size=[30,30])#随机生成一个整数矩阵
x,y=data[0],data[1]
fig,ax=plt.subplots()
ax.scatter(x[:10],y[:10],'r')
ax.scatter(x[10:20],y[10:20],'b')
ax.set_xlabel('x')
ax.set_ylabel('y')
plt.show()

2.三维散点图

from mpl_toolkits.mplot3d import Axes3D
x,y,z=data[0],data[1],data[2]
ax=plt.subplot(111,projection='3d')
ax.scatter(x[:10],y[:10],c='r')
ax.scatter(x[10:20],y[10:20],c='b')
ax.scatter(x[20:30],y[20:30],c='g')
plt.show()