diff --git a/fluents/plots.py b/fluents/plots.py index 0d5f931..1d61e74 100644 --- a/fluents/plots.py +++ b/fluents/plots.py @@ -349,6 +349,8 @@ class ScatterPlot(Plot): """The ScatterPlot is slower than scattermarker, but has size option.""" @plotlogger def __init__(self, dataset_1, dataset_2, id_dim, sel_dim, id_1, id_2, c='b', s=30, sel_dim_2=None, name="Scatter plot", **kw): + """Initializes a scatter plot. + """ Plot.__init__(self, name) self.dataset_1 = dataset_1 @@ -407,6 +409,7 @@ class ScatterPlot(Plot): self.yaxis_data, alpha=0, c='w',s=self.s, + edgecolor='r', linewidth=0, zorder=4) self._background = self.canvas.copy_from_bbox(self.axes.bbox) @@ -480,7 +483,7 @@ class ScatterPlot(Plot): linewidth = scipy.zeros(self.xaxis_data.shape, 'f') index = self.get_index_from_selection(self.dataset_1, selection) if len(index) > 0: - linewidth.put(1, index) + linewidth[index] = 1.5 self.selection_collection.set_linewidth(linewidth) if self._use_blit and len(index)>0 : @@ -527,9 +530,14 @@ class HistogramPlot(Plot): self.current_dim = dataset.get_dim_name(1) if dataset.shape[1]==1: self.current_dim = dataset.get_dim_name(0) + + # Set default paramteters + if not kw.has_key('bins'): + kw['bins'] = 20 + # Initial draw self.axes.grid(False) - bins = min(self._data.size, 20) + bins = min(self._data.size, kw['bins']) count, lims, self.patches = self.axes.hist(self._data, bins=bins) # Add identifiers to the individual patches