matplotlib基础（1）之散点图scatter

亦凉 2022-09-04 10:56 203阅读 0赞

1.首先安装matplotlib

pip install matplotlib

2.引入matplotlib

import matplotlib.pyplot as plt
    import numpy as np

3.matplotlib散点图

#  散点图  scatter(x ,y ,...)  x y 为数组
    
    # uniform   均匀分布随机采样
    arr1 = np.random.uniform(0.0, 5.0, size=(100, 2))  # 100行 2列的元组
    print(arr1)
    
    arr2 = np.random.uniform(4.2, 10, size=(100, 2))  # 100行 2列的元组
    print(arr2)
    
    
    # 保存数据
    np.savetxt('./matplot_test_data1.txt', arr1, fmt='%f %f',
               delimiter='\n')  # fmt 格式小数点    delimiter分隔符
    np.savetxt('./matplot_test_data2.txt', arr2, fmt='%f %f',
               delimiter='\n')  # fmt 格式小数点    delimiter分隔符
    
    # 读取数据
    a = np.loadtxt('./matplot_test_data1.txt')
    b = np.loadtxt('./matplot_test_data2.txt')
    
    """
     [
        [9.2991324  7.36309413]
        ...
        [6.70664294 5.11371024]
        [8.54146186 8.70348462]
        [7.00828175 7.30186039]
        [4.92537553 9.89860809]
        [5.58266787 8.5275205 ]
     ]
    """
    # 绘图   散点图
    # scatter ( 数组 )  a[: , 0] （ : => 取a的所有的行 ， 0=> 取a的第一列 ）
    plt.scatter(a[:, 0], a[:, 1])
    plt.scatter(b[:, 0], b[:, 1])
    
    print(a.shape)  # 100行 列
    
    x = [1, 2, 3, 4]
    y = [2, 3, 4, 8]
    # s:点的大小  color:点的颜色缩写  marker:点的样式
    
    plt.scatter(x, y, s=[20, 40, 80, 300], color=['r', 'g', 'b', 'y'], marker='o')
    
    # 练习  男人，女人身高 散点图
    men = np.random.uniform(150, 200, size=20)
    men_id = np.arange(21)  # [0-20]
    men_id = np.delete(men_id, 0)  # [1-20]
    
    # 要将men, men_id 合并为二维数组
    # 1. 将men变形
    men = men.reshape(20, 1)  # 变为20行，一列
    men_id = men_id.reshape(20, 1)  # 变为20行，一列
    men_male = np.concatenate((men_id, men), axis=1)  # 列相加
    
    
    women = np.random.uniform(130, 180, size=20)
    women_id = np.arange(21)  # [0-20]
    women_id = np.delete(women_id, 0)  # [1-20]
    
    # 要将men, men_id 合并为二维数组
    # 1. 将men变形
    women = women.reshape(20, 1)  # 变为20行，一列
    women_id = women_id.reshape(20, 1)  # 变为20行，一列
    women_male = np.concatenate((women_id, women), axis=1)  # 列相加
    # print(women_male)
    
    
    x2 = np.concatenate((men_male, women_male))
    
    y2 = np.concatenate((np.ones(20), np.zeros(20)))  # 前20个代表男，和X对应
    # 查 x 数据中男性的数据   x[y==1]   x[y==1,0]第一列
    
    # 男性的
    plt.scatter(x2[y2 == 1, 0], x2[y2 == 1, 1], color='r')
    
    # 女性
    plt.scatter(x2[y2 == 0, 0], x2[y2 == 0, 1], color='g')
    
    plt.show() #只会输出图表，不会输出对象信息

4.效果图

在vsCode中安装matplotlib插件

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2Nhb3lhbjA4Mjk_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2Nhb3lhbjA4Mjk_size_16_color_FFFFFF_t_70]: /images/20220829/2457f43a517643baae1a1853f0996753.png