import numpy
import matplotlib.pyplot as plt
plt.style.use("ggplot")
data = numpy.load("diffusion_map_dataset.npy")
plt.scatter(data[:167, 0], data[:167, 1])
plt.scatter(data[167:334, 0], data[167:334, 1])
plt.scatter(data[334:, 0], data[334:, 1])
import tikzplotlib
data = numpy.load("diffusion_map_embedding.npy")
plt.scatter(data[:167, 0], data[:167, 1])
plt.scatter(data[167:334, 0], data[167:334, 1])
plt.scatter(data[334:, 0], data[334:, 1])
print("idx,x1,y1,x2,y2,x3,y3")
for a in enumerate(zip(data[:167], data[167:334], data[334:])):
print("{}, {}, {}, {}, {}, {}, {}".format(a[0], a[1][0][0], a[1][0][1], a[1][1][0], a[1][1][1], a[1][2][0], a[1][2][1]))