как построить линии потока, когда я знаю u и v компоненты скорости (numpy 2d arrays), используя программу черчения в python?

Надеюсь, название было достаточно понятным, я решаю задачу 2D lid-driven cavity(square domain) методом дробного шага, конечно-разностной формулировкой (Navier-Stokes primitive variable form), я получил u и v компоненты скорости по всей области, без ручного расчета обтекания, есть ли команда или инструмент для черчения, который делает эту работу за меня?

Надеюсь, этот вопрос достаточно актуален для программирования, так как мне нужен инструмент для черчения обтекания без явного расчета.

Я решил ту же задачу в форме NS для потока-вихря, мне просто пришлось построить контурный график функции потока, чтобы получить линии потока.

Я надеюсь, что инструмент или плоттер является библиотекой python, и более того, может быть установлен в fedora (я могу пойти на компромисс и использовать mint) без особых проблем!!!

Я буду благодарен, если кто-нибудь укажет библиотеку и соответствующую команду (это сэкономит много времени)