Commit f1b4835f authored by Elijah Andrews's avatar Elijah Andrews

Improved 2D point set plotting.

parent 831835f0
......@@ -98,12 +98,12 @@ def plot_3d_points(points, c=None, cmap_name="RdBu", center_cmap=True):
def plot_2d_point_sets(point_sets):
colors = ["r", "b", "g", "orange", "k", "yellow"]
fig = plt.figure()
ax = fig.add_subplot(111) # type: Axes3D
ax = fig.add_subplot(111, aspect='equal')
for i in range(len(point_sets)):
points = np.array(point_sets[i])
ax.scatter(points[:, 0], points[:, 1], c=colors[i])
lim = [min([ax.get_xlim()[0], ax.get_ylim()[0]]),
max([ax.get_xlim()[1], ax.get_ylim()[1]])]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment