python - Python 3 中数据的图形选择

Question

要解决的问题

我有两个 2D numpy 数组。其中一个是 float 组，另一个是字符串数组。每个 float 组元素都是从文件中提取的，其名称位于相应的字符串数组元素中。

我想绘制 float 组的二维热图，颜色与数组元素的大小相对应。

基于此，我想以交互方式选择一个区域(例如使用多边形或套索工具)。我希望将已选择的数组的索引写入列表/数组，以便可以提取相应的文件名以进行进一步处理。

我使用的是 Python 3。

我花了几个小时试图在 Bokeh 中解决这个问题，但没有成功。

我的问题如下

哪些 Python 库最适合这个问题？

鉴于上述库，您有什么入门技巧吗？

非常感谢。

注意:我一直在使用 Python 进行一些科学数据编程，但不认为自己是一个经验丰富的程序员。

Answer 1

没有简单的方法可以做到这一点，但是使用 matplotlib 就可以做到这一点。 Matplotlib 有两个类 SpanSelector 和 LassoSelector 可供您使用。您可以找到一些文档 here 。 Here是 LassoSelector 的示例。