set of boundaries for get_dofs in ex05

kinnala · Nov 14, 2021 · d3b1787 · d3b1787
1 parent 334bec5
commit d3b1787
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/docs/examples/ex02.py b/docs/examples/ex02.py
@@ -101,11 +101,8 @@ def C(T):
 K = asm(bilinf, ib)
 f = 1e6 * asm(unit_load, ib)
 
-D = np.concatenate(
-    [
-        ib.get_dofs("left").facet["u_n"],
-        *[ib.get_dofs(s).nodal["u"] for s in ["left", "right", "top"]],
-    ]
+D = np.hstack(
+    [ib.get_dofs("left").all("u_n"), ib.get_dofs({"left", "right", "top"}).all("u")]
 )
 
 x = solve(*condense(K, f, D=D))