python - Insert image in matplotlib legend -


i'd insert couple small graphics (vector graphics can made raster if necessary) legend of maplotlib plot. there 1 graphic per item in legend.

i know manually draw entire legend using something annotation box looks tedious, , small change in figure require fixing hand.

is there way include graphics in label in call pyplot.plot or later in pyplot.legend call?

so, below little hacky, can of way there. note: need replace [path image] image want (otherwise grace hopper free!). can make image larger default passing image_stretch parameter. hacky way fix aspect ratio on image. use labelspacing parameter if images overlap 1 series next.

import os  matplotlib.transforms import transformedbbox matplotlib.image import bboximage matplotlib.legend_handler import handlerbase matplotlib._png import read_png  class imagehandler(handlerbase):     def create_artists(self, legend, orig_handle,                        xdescent, ydescent, width, height, fontsize,                        trans):          # enlarge image these margins         sx, sy = self.image_stretch           # create bounding box house image         bb = bbox.from_bounds(xdescent - sx,                               ydescent - sy,                               width + sx,                               height + sy)          tbb = transformedbbox(bb, trans)         image = bboximage(tbb)         image.set_data(self.image_data)          self.update_prop(image, orig_handle, legend)          return [image]      def set_image(self, image_path, image_stretch=(0, 0)):         if not os.path.exists(image_path):             sample = get_sample_data("grace_hopper.png", asfileobj=false)             self.image_data = read_png(sample)         else:             self.image_data = read_png(image_path)          self.image_stretch = image_stretch  # random data x = np.random.randn(100) y = np.random.randn(100) y2 = np.random.randn(100)  # plot 2 series of scatter data s = plt.scatter(x, y, c='b') s2 = plt.scatter(x, y2, c='r')  # setup handler instance scattered data custom_handler = imagehandler() custom_handler.set_image("[path image]",                          image_stretch=(0, 20)) # grace hopper  # add legend scattered data, mapping # scattered points custom handler plt.legend([s, s2],            ['scatters 1', 'scatters 2'],            handler_map={s: custom_handler, s2: custom_handler},            labelspacing=2,            frameon=false)

here's produces:

grace hopper


Comments

Post a Comment

Popular posts from this blog

Python Kivy ListView: How to delete selected ListItemButton? -

Image
i'm trying learn kivy building simple todo-list app suggested dusty phillips, author of book "creating apps in kivy". this code far: from kivy.app import app kivy.uix.boxlayout import boxlayout kivy.properties import objectproperty kivy.uix.listview import listitembutton class taskbutton(listitembutton): pass class todoroot(boxlayout): task_input = objectproperty() task_list = objectproperty() def add_task(self): self.task_list.adapter.data.extend([self.task_input.text]) self.task_list._trigger_reset_populate() def del_task(self): pass class todoapp(app): def build(self): return todoroot() if __name__ == '__main__': todoapp().run() and kv file: #: import main todo #: import listadapter kivy.adapters.listadapter.listadapter #: import listitembutton kivy.uix.listview.listitembutton todoroot: <todoroot>: orientation: "vertical" task_input: task_input_view t...
Read more

ruby - How do I merge two hashes into a hash of arrays? -

http://codepad.org/wmyzqlis i'd merge these 2 arrays: a = { :a => 'a', :b => 'b', :d => 'd' } b = { :a => '1', :b => 'b', :c => 'c' } into hash looks this: {:a => ["a", "1"], :b => ["b", "b"], :c => [nil, "c"], :d => ["d", nil] } this doesn't work: p a.merge(b) { |k, v1, v2| [v1, v2] } # {:c=>"c", :a=>["a", "1"], :b=>["b", "b"], :d=>"d"} it's because hash#merge call block duplicate keys. a = { :a => 'a', :b => 'b', :d => 'd' } b = { :a => '1', :b => 'b', :c => 'c' } hash[(a.keys|b.keys).map { |key| [key, [a.fetch(key, nil), b.fetch(key, nil)]] }] # => {:a=>["a", "1"], :b=>["b", "b"], :d=>["d", nil], :c=>...
Read more