Python中数据的图形选择3

时间:2016-11-09 20:42:15

标签: python numpy plot interactive

要解决的问题

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

我想绘制浮点数组的2D热图,颜色对应于数组元素的大小。

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

我使用的是Python 3。

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

我的问题如下

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

鉴于上述图书馆,您有任何入门提示吗?

非常感谢。

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

1 个答案:

答案 0 :(得分:1)

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