def saveImage(points,data,fname): x,y,data_2d = np.real(tools.reshape_to_2d_plane(points,data[0],eval_dims,'xy')) mask = x**2+y**2<1 data_2d[mask] = np.nan plt.imshow(data_2d.T,extent = [-3,3,-3,3],vmin=-.4,vmax=.4) plt.savefig(fname) plt.clf()
def saveImage(points, data, fname): field = data[0, ...] ** 2 + data[1, ...] ** 2 + data[2, ...] ** 2 x, y, data_2d = np.real(tools.reshape_to_2d_plane(points, field, eval_dims, "xy")) mask = x ** 2 + y ** 2 < 1 data_2d[mask] = np.nan plt.imshow(data_2d.T, extent=[-3, 3, -3, 3], vmin=0, vmax=2) plt.colorbar() plt.savefig(fname) plt.clf()