[BUG]: vtk_mesh on subset of cells based of function space

[jorgensd]

Summarize the issue

Currently broken, as shown in: https://fenicsproject.discourse.group/t/question-about-selecting-boundaries-for-boundary-conditions/16865/6?u=dokken

This line should be changed: https://github.com/FEniCS/dolfinx/blob/main/python/dolfinx/plot.py#L134 using entities to access the correct rows

How to reproduce the bug

See example on discourse, will make smaller example when on laptop

Minimal Example (Python)

Output (Python)

Version

main branch

DOLFINx git commit

No response

Installation

No response

Additional information

No response