03.简易散点图

import unittest
import matplotlib.pyplot as plt

"""简易散点图"""


class MyTestCase(unittest.TestCase):
    def test_something(self):
        self.assertEqual(True, False)

    def test_o_01(self):
        """不同坐标类型"""
        i = 0
        for marker in [\'o\', \'.\', \',\', \'X\', \'x\', \'+\', \'v\',
                       \'^\', \'<\', \'>\', \'s\', \'d\', \'8\', \'4\',
                       \'3\', \'2\', \'1\', \'P\', \'p\', \'H\', \'h\',
                       \'*\', \'_\', \'|\', \'d\', \'D\']:
            plt.plot([i], [i], marker, label="marker=\'{0}\'".format(marker))
            i += 1
            plt.legend()
            plt.xlim(-1, 50)
            plt.ylim(-1, 50)

        plt.savefig(\'test_o_01.png\')

    def test_o_02(self):
        """不同图标混合使用"""
        plt.plot([1, 2, 3], [4, 5, 6], \'-or\')  # 直线，圆圈，红色

        plt.savefig(\'test_o_02.png\')

    def test_o_03(self):
        """自定义线条和散点属性"""
        plt.plot([1, 2, 3], [4, 5, 6], \'-p\', color=\'gray\',
                 markersize=15, linewidth=4,
                 markerfacecolor=\'yellow\',  # 标记颜色
                 markeredgecolor=\'red\',  # 标记边缘颜色
                 markeredgewidth=2)

        plt.savefig(\'test_o_03.png\')

    def test_o_04(self):
        """使用scatter画散点图
              cmap https://matplotlib.org/tutorials/colors/colormaps.html
        """
        plt.scatter([1, 2, 3], [4, 5, 6], c=[100, 120, 150], s=[100, 60, 30], alpha=0.3, cmap=\'viridis\')
        plt.colorbar()  # 显示颜色条
        plt.savefig(\'test_o_04.png\')


if __name__ == \'__main__\':
    unittest.main()

test_o_01.png

test_o_02.png

test_o_03.png

test_o_04.png