pyAnsys_DPF_example_3.py

def get_grid_with_field(meshed_region, field):
    name = '_'.join(field.name.split("_")[:-1])
    location = field.location
    if location == dpf.locations.nodal:
        mesh_location = meshed_region.nodes
    elif location == dpf.locations.elemental:
        mesh_location = meshed_region.elements
    else:
        raise ValueError(
            "Only elemental or nodal location are supported for plotting."
        )
    overall_data = np.full(len(mesh_location), np.nan)
    ind, mask = mesh_location.map_scoping(field.scoping)
    overall_data[ind] = field.data[mask]
    grid = meshed_region.grid
    if location == dpf.locations.nodal:
        grid.point_data[name] = overall_data
    elif location == dpf.locations.elemental:
        grid.cell_data[name] = overall_data
    
    return grid