Python 中数据的图形选择 3

Question

要解决的问题

我有两个二维 numpy 数组。其中一个是浮点数组，另一个是字符串数组。每个 float 数组元素都是从文件中提取的，其名称在相应的字符串数组元素中。

我想绘制浮点数组的二维热图，颜色对应于数组元素的大小。

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

我正在使用Python 3.

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

我的问题如下

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

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

非常感谢。

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

Answer 1

没有简单的方法可以做到这一点，但是使用 matplotlib 非常可行。 Matplotlib 有两个 类 SpanSelector 和 LassoSelector，你可以 use.You 可以找到一些文档 here. Here 是 LassoSelector 的一个例子。