<s> import numpy as np <EOL> def plot_grid ( x , cmap = None , border_color = None , axes = None , linewidth = <NUM_LIT:1> ) : <EOL> """<STR_LIT>""" <EOL> import matplotlib . pyplot as plt <EOL> if axes is not None : <EOL> axes . matshow ( x , cmap = cmap ) <EOL> else : <EOL> axes = plt . matshow ( x , cmap = cmap ) . get_axes ( ) <EOL> axes . set_xticks ( np . arange ( <NUM_LIT:1> , x . shape [ <NUM_LIT:1> ] ) - <NUM_LIT> ) <EOL> axes . set_yticks ( np . arange ( <NUM_LIT:1> , x . shape [ <NUM_LIT:0> ] ) - <NUM_LIT> ) <EOL> if border_color is None : <EOL> border_color = '<STR_LIT>' <EOL> axes . grid ( linestyle = "<STR_LIT:->" , linewidth = linewidth , color = border_color ) <EOL> axes . set_xticklabels ( ( ) ) <EOL> axes . set_yticklabels ( ( ) ) <EOL> return axes </s>
