Ejemplos de reshape_to_2d_plane en Python

Ejemplo n.º 1

Archivo: acoustic_example.py Proyecto: rorypeck/tdbempp

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()

Ejemplo n.º 2

Archivo: maxwell_example.py Proyecto: rorypeck/tdbempp

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()