1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
from collections import namedtuple
import matplotlib.pyplot as plot
class Style(namedtuple('SS', ['marker', 'line', 'color'])):
def fmt(self):
return self.marker + self.line + self.color
class StyleMap(object):
'''Maps configs (dicts) to specific line styles.'''
DEFAULT = Style('.', '-', 'k')
def __init__(self, col_list, col_values):
'''Assign (some) columns in @col_list to fields in @Style to vary, and
assign values for these columns to specific field values.'''
self.value_map = {}
self.field_map = {}
for field, values in self.__get_all()._asdict().iteritems():
next_column = col_list.pop(0)
value_dict = {}
for value in sorted(col_values[next_column]):
value_dict[value] = values.pop(0)
self.value_map[next_column] = value_dict
self.field_map[next_column] = field
def __get_all(self):
'''A Style holding all possible values for each property.'''
return Style(list('.,ov^<>1234sp*hH+xDd|_'), # markers
['-', ':', '--'], # lines
list('bgrcmyk')) # colors
def get_style(self, kv):
'''Translate column values to unique line style.'''
style_fields = {}
for column, values in self.value_map.iteritems():
if column not in kv:
continue
field = self.field_map[column]
style_fields[field] = values[kv[column]]
return StyleMap.DEFAULT._replace(**style_fields)
def get_key(self):
'''A visual description of this StyleMap.'''
key = []
for column, values in self.value_map.iteritems():
# print("***%s, %s" % column, values)
for v in values.keys():
sdict = dict([(column, v)])
style = self.get_style(sdict)
styled_line = plot.plot([], [], style.fmt())[0]
description = "%s:%s" % (column, v)
key += [(styled_line, description)]
return sorted(key, key=lambda x:x[1])
|
