numpy和matplotlib生成图片

时间:2024-03-19 14:02:10

一.numpy和matplotlib生产图片,图片原理,图片有三色彩,rgb即为RED,BLUE,GREEN。三原色。

首先生成一个numpy数组

import numpy as np
import matplotlib.pylab as plt
import pandas as pd
import random
import cv2
#图形数据处理
class ml:
    
    def __init__(self):
        print("cnn test")
        self.path=''
        
    def show_im(self,tmp_im):
        plt.imshow(tmp_im)
        plt.axis('off')
        plt.show()
        
    def read_file(self):
        #初始值为1的得值 分别代表第一层数据为200,第二层为300,第三层为3,决定图片颜色得最里层数据长度为3,分别代表三种颜色
        im=np.ones((200,300,3), dtype='int8')
        size=im.shape
        #修改类型
        im=im.astype('int8')
        print('-----------')
        #绘图,这里修改四处地方,是一个王字。
        for b in range(size[1]):
            for a in range(size[0]):
                im[a][b]=[127,127,127]
                if a in [0,1,2,3]:
                    if b<90 and b>10:
                    	#设置颜色
                        im[a][b]=[125,0,0]
                    
                if a in [50,51,52,53]:
                    if b<80 and b>20:
                        im[a][b]=[125,0,0]
                if a in [100,101,102,103]:
                    if b<100:
                        im[a][b]=[125,0,0]
                        
                if b in [50,51,52,53]:
                    if a <100:
                        im[a][b]=[125,0,0]             
        print(im)
        #显示图片
        self.show_im(im)
        
    def save_data(self):
        print(1)  
        
if __name__=='__main__':
    obj=ml()
    obj.read_file()

图片效果

numpy和matplotlib生成图片