#! / usr / bin / env python # - * - кодировка: utf-8 - * -import sys import numpy as np import matplotlib.pyplot as plt from matplotlib import rcParamsrcParams [ 'font.size' ] = 18# Доступные методы интерполяции: # 'none', 'ближайший', 'билинейный', 'бикубический', 'spline16', # 'spline36', 'hanning', 'Hamming', 'hermite', 'kaiser', 'quadric' , # 'catrom', 'гауссовский', 'bessel', 'mitchell', 'sinc', 'lanczos'МЕТОДЫ = [ 'ближайшего' , 'билинейной' , 'бикубического' , 'Эрмит ] ФОРМАТЫ = [ 'PNG' , 'PDF' , 'SVG' ] ЦВЕТ = 'виридис'N = 5 нп . случайный . семя ( 1 ) сетка = np . arange ( 0 , N , 1 ) данные = np . round ( np . random . rand ( N , N ), 1 ) сетка = np . meshgrid ( сетка , сетка )для Interp в СПОСОБОВ : fig = plt . рисунок ( figsize = ( 5 , 5 )) топор = рис . add_axes ([ 0,125 , 0,175 , 0,75 , 0,75 ]) plt . imshow ( данные , интерполяция = интерполяция , CMAP = ЦВЕТ , Vmin = 0 , Vmax = 1 ) PLT . сюжет ( сетка [ 0 ], сетка [ 1 ], маркер = '.' , мс = 8, цвет = 'k' , lw = 0 ) plt . title ( interp , weight = 'bold' ) plt . xlim ( сетка . min () - 0,5 , сетка . max () + 0,5 ) plt . ylim ( сетки . мин () - 0,5 , сетка . макс () + 0,5 ) PLT. xticks ( сетка ) plt . yticks ( сетка ) cax = рис . add_axes ([ 0,125 , 0,075 , 0,75 , 0,03 ]) cb = plt . Colorbar ( CAX = CAX , ориентация = 'горизонтальные' , клещами = пр . LINSPACE ( 0 , 1 , 6 )) CB . твердые тела . set_edgecolor ( 'лицо' ) для доп. в ФОРМАТАХ : plt . savefig ( " {} _ {} . {} " . формат ( sys . argv [ 0 ], interp , ext ))