Fix dofmap extraction for subsets of cells. Resolve #3607 (#3615)

FEniCS · Feb 5, 2025 · d45b65f · d45b65f
1 parent 4ac3bc1
commit d45b65f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/dolfinx/plot.py b/python/dolfinx/plot.py
@@ -130,6 +130,5 @@ def _(V: fem.FunctionSpace, entities=None):
     topology = np.zeros((len(entities), num_dofs_per_cell + 1), dtype=np.int32)
     topology[:, 0] = num_dofs_per_cell
     dofmap_ = dofmap.list
-
-    topology[:, 1:] = dofmap_[: len(entities), perm]
+    topology[:, 1:] = dofmap_[entities][:, perm]
     return topology.reshape(1, -1)[0], cell_types, V.tabulate_dof_coordinates()