diff --git a/samila/genimage.py b/samila/genimage.py index 0d7c24a..02fe796 100644 --- a/samila/genimage.py +++ b/samila/genimage.py @@ -48,6 +48,7 @@ class GenerativeImage: def plot( self, color=DEFAULT_COLOR, + bgcolor=DEFAULT_BACKGROUND_COLOR, spot_size=DEFAULT_SPOT_SIZE, size=DEFAULT_IMAGE_SIZE, projection=DEFAULT_PROJECTION): @@ -56,6 +57,8 @@ class GenerativeImage: :param color: point colors :type color: str + :param bgcolor: background color + :type bgcolor: str :param spot_size: point spot size :type spot_size: float :param size: figure size @@ -66,6 +69,15 @@ class GenerativeImage: """ fig = plt.figure() fig.set_size_inches(size[0], size[1]) + fig.set_facecolor(bgcolor) ax = fig.add_subplot(111, projection=projection) - ax.scatter(self.data2, self.data1, alpha=0.1, c=color, s=spot_size) - ax.axis('off') + ax.set_facecolor(bgcolor) + ax.scatter( + self.data2, + self.data1, + alpha=0.1, + edgecolors=color, + s=spot_size) + ax.set_axis_off() + ax.patch.set_zorder(-1) + ax.add_artist(ax.patch) diff --git a/samila/params.py b/samila/params.py index 17cb286..404a610 100644 --- a/samila/params.py +++ b/samila/params.py @@ -6,6 +6,7 @@ DEFAULT_START = -1 * math.pi DEFAULT_STOP = math.pi DEFAULT_STEP = 0.01 DEFAULT_COLOR = "black" +DEFAULT_BACKGROUND_COLOR = "white" DEFAULT_IMAGE_SIZE = (10, 10) DEFAULT_SPOT_SIZE = 0.01 DEFAULT_PROJECTION = None