import matplotlib.pyplot def flat_landscape(size): environment = [] for i in range(size): row_list = [] for j in range(size): row_list.append(50) environment.append(row_list) return environment def sloping_x_landscape(size): environment = [] for i in range(size): row_list = [] for j in range(size): row_list.append(j) environment.append(row_list) return environment def sloping_both_landscape(size): environment = [] for i in range(size): row_list = [] for j in range(size): row_list.append(j + i) environment.append(row_list) return environment size = 300 landscape = sloping_both_landscape(size) matplotlib.pyplot.imshow(landscape) matplotlib.pyplot.show()