Python 绘制散点图

matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！

import numpy as np
import matplotlib.pyplot as plt

if __name__ == "__main__":

    coords = np.load(file="coords.npy") #(n,3)
    x = coords[:,0]
    y = coords[:,1]
    z = coords[:,2]
   
   # 给不同的点赋不同的颜色
    colors = []
    for i in range(x.shape[0]):
        if 满足某个条件:
            colors.append((255/255, 255/255, 255/255))
        else:
            colors.append((0, 0, 0))


    ax = plt.subplot(projection = '3d')  # 创建工程
    ax.set_title('3d_point_show')  # 设置图名
    ax.scatter(x, y, z, c = colors)   # 绘制数据点 c: 'r'红色，'y'黄色，等颜色
    # 也可以直接设置为统一的颜色
    # ax.scatter(x, y, z, c = 'r')
    ax.set(xlabel="X", ylabel="Y", zlabel="Z")
    plt.savefig('3D.jpg', bbox_inches='tight', dpi=2400) #保存
    plt.show()

matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！ matplotlib 是一个绘制三维散点图非常有用的库，下面的代码即基于一组数据绘制散点图的过程！