[ Python ] matplotlib

標籤：ace   index   pytho   res   result   int   param   end   print   

# -*- coding: utf-8 -*-import unittestclass TestClass1(unittest.TestCase):    @unittest.skip("core.ok")    def test_tuple(self):        data = (11,22,33)        print(data)        x,y,z = data        print(x,y,z)    @unittest.skip("matplotlib.screen")    def test_scrap_screen(self):        from PIL import Image, ImageGrab        import numpy as np        from matplotlib import pyplot as plt        sim = ImageGrab.grab()        w,h = sim.size        del(sim)        dpi = plt.rcParams[‘figure.dpi‘]        print("screen size: %d, %d" % (w, h))        print("screen inch: %2.2f, %2.2f" %(w/dpi, h/dpi))        djx = np.sqrt(w*w + h*h)        print("screen djx = %d" % djx)        print("screen dpi = %f" % (djx/15.4))        print("screen size= %d, %d" % (w/221, h/221))    # @unittest.skip("xxxyyyxxxyyyyy")    def test_full_screen(self):        from matplotlib import pyplot as plt        from PIL import Image        img_name = "/Users/jacobzhao/Downloads/sunyunzhu1.jpeg"        img = Image.open(img_name)        print("dip = %d" % plt.rcParams[‘figure.dpi‘])        plt.figure()        plt.axis(‘off‘)        ax = plt.gca()        ax.spines[‘top‘].set_visible(False)        leg = plt.legend()        leg.get_frame().set_linewidth(0.0)        plt.subplots_adjust(hspace=0, wspace=0)        plt.imshow(img)        plt.show()if __name__ == ‘__main__‘:    unittest.main()    # suite = unittest.TestSuite()    # suite.addTest(TestClass1(‘test_scrap_screen‘))    # runner = unittest.TextTestResult()    # runner.run(suite)

　　

 

# -*- coding: utf-8 -*-from functools import reduceimport numpy as npimport matplotlib.pyplot as pltfrom PIL import Imageclass MyImage(object):    def __init__(self, filepath):        self.filepath = filepath        self.imgfile = Image.open(filepath)        if self.imgfile.mode != ‘RGB‘:            self.imgfile = self.imgfile.convert("RGB")        self.imgdata = np.array(self.imgfile)        self.info = {            "mode": self.imgfile.mode,            "shape":self.imgdata.shape        }    def showImg(self, title=‘NO-Title‘):        plt.figure(title)        plt.imshow(self.imgdata)        plt.axis(‘off‘)        plt.show()    def printInfo(self):        print(self.info)class MyImage2(object):    def __init__(self, filepath1, filepath2):        self.filepath1 = filepath1        self.filepath2 = filepath2        self.imgfile1 = Image.open(filepath1)        self.imgfile2 = Image.open(filepath2)        if self.imgfile1.mode != ‘RGB‘:            self.imgfile1 = self.imgfile1.convert("RGB")        if self.imgfile2.mode != ‘RGB‘:            self.imgfile2 = self.imgfile2.convert("RGB")        self.imgdata1 = np.array(self.imgfile1)        self.imgdata2 = np.array(self.imgfile2)        self.info1 = {            "mode": self.imgfile1.mode,            "shape":self.imgdata1.shape        }        self.info2 = {            "mode": self.imgfile2.mode,            "shape":self.imgdata2.shape        }    def showImg(self, title1=‘Image-1‘, title2=‘Image-2‘):        fig = plt.figure(figsize=(10, 5))        ax1 = fig.add_subplot(1, 2, 1)        ax1.imshow(self.imgdata1)        ax1.axis(‘off‘)        plt.title(title1)        ax2 = fig.add_subplot(1, 2, 2)        ax2.imshow(self.imgdata2)        ax2.axis(‘off‘)        plt.title(title2)        # plt.suptitle("Image-Test")        plt.show()    def printInfo(self):        print(self.info1)        print(self.info2)class MyImageN(object):    def __init__(self, filepaths):        self.size = len(filepaths)        pos = list(range(self.size))        self.filepaths = filepaths        self.imgfiles = []        self.imgdatas = []        self.imginfos = []        for index, filepath in zip(pos, filepaths):            info={‘title‘:‘img-‘+str(index)}            self.imgfiles.append(Image.open(filepaths[index]))            info[‘mode‘] = self.imgfiles[index].mode            if self.imgfiles[index].mode != ‘RGB‘:                self.imgfiles[index] = self.imgfiles[index].convert(‘RGB‘)            self.imgdatas.append(np.array(self.imgfiles[index]))            info[‘shape‘] = self.imgdatas[index].shape            self.imginfos.append(info)    def showImg(self, titles = None):        fig = plt.figure(figsize=(4*self.size, 4))        for index in range(self.size):            ax = fig.add_subplot(1, self.size, index + 1)            ax.imshow(self.imgdatas[index])            ax.axis(‘off‘)            plt.title("Image-"+str(index))        # plt.suptitle("Image-Test")        plt.show()    def printInfos(self):        for index in range(self.size):            print(index+1, ‘->‘, self.imginfos[index])    def processImage(self, index, func):        curr_data = self.imgdatas[index]        self.imgdatas[index] = func(self.imgfiles[index])def picGray(imgfile):    return imgfile.convert(‘L‘)if __name__ == ‘__main__‘:    syz = [        r"/Users/jacobzhao/Downloads/sunyunzhu1.jpeg",        r"/Users/jacobzhao/Downloads/sunyunzhu2.png",        # r"/Users/jacobzhao/Downloads/sunyunzhu3.png",    ]    # MyImage(syz[0]).showImg()    # MyImage(syz[1]).showImg()    syzImgs = MyImageN(syz)    # syzImgs.showImg()    syzImgs.processImage(0, picGray)    syzImgs.processImage(1, picGray)    # syzImgs.processImage(2, picGray)    syzImgs.printInfos()    syzImgs.showImg()

　　

 

# -*- coding: utf-8 -*-import matplotlib.pyplot as pltimport numpy as npfig = plt.figure(figsize=(8, 4))ax1 = fig.add_subplot(2,2,1)ax2 = fig.add_subplot(2,2,2)ax3 = fig.add_subplot(2,1,2)ax3.plot(np.random.randn(50).cumsum(), ‘k--‘)ax1.hist(np.random.randn(100), bins=10, color=‘b‘, alpha=0.3)ax2.scatter(np.arange(30), np.arange(30) + 3*np.random.randn(30))plt.show()a = [x+11 for x in range(6)]b = [y+101 for y in range(6)]print(a, b)for m,n in zip(a,b):    print(m, n)print(list(range(8)))

　　

[ Python ] matplotlib