Графический Выбор данных в Python 3

Question

Проблема Решить

У меня есть два 2D Numpy массивы. Один из них представляет собой массив поплавков, другой - массив строк. Каждый элемент массива float извлекается из файла с именем в соответствующем элементе массива строк.

Я хочу построить двумерную карту памяти массива поплавков с цветом, соответствующим величине элемента массива.

Основываясь на этом, я хотел бы интерактивно выбирать область (например, с помощью инструмента многоугольника или лассо). Я хочу, чтобы индексы массива, которые были выбраны, затем были записаны в список/массив, поэтому соответствующие имена файлов могут быть извлечены для дальнейшей обработки.

Я использую Python 3.

Я потратил несколько часов, пытаясь добиться успеха с этой проблемой в Bokeh, но не имели никакого успеха.

Мои вопросы являются следующие

Какие библиотеки Python наилучшим образом подходит эта проблема?

Учитывая перечисленные выше библиотеки (ы), есть ли у вас какие-либо советы для начала?

Большое спасибо.

Примечание: Я использую Python для некоторых научных программ данных, но не считаю себя опытным программистом.

Answer 1

Нет простого способа сделать это, однако это вполне выполнимо с помощью matplotlib. Matplotlib имеет два класса SpanSelector и LassoSelector, которые вы можете использовать. Вы можете найти документацию here. Here - пример с LassoSelector.