import numpy as np 
import matplotlib.pyplot as plt
data = np.load('data/wgan-gp.npy')
def plt_plot(data):
    data = data.reshape(-1,3,60)
    for i in range(len(data)):
        plt.plot(data[i][0][0:30],data[i][0][30:60],color='red')
        plt.plot(data[i][1][0:30],data[i][1][30:60],color='blue')
        plt.plot(data[i][2][0:30],data[i][2][30:60],color='green')
        plt.xlim(0.,1.)
        plt.ylim(0.,1.)
        plt.xticks(np.arange(0,1,0.1))
        plt.yticks(np.arange(0,1,0.1))
        plt.xticks(np.arange(0,1,0.1))
        plt.yticks(np.arange(0,1,0.1))
        plt.show()
def plt_scatter(data):
    data = data.reshape(-1,3,60)
    for i in range(len(data)):
        plt.scatter(data[i][0][0:30],data[i][0][30:60],color='red')
        plt.scatter(data[i][1][0:30],data[i][1][30:60],color='blue')
        plt.scatter(data[i][2][0:30],data[i][2][30:60],color='green')
        plt.xlim(0.,1.)
        plt.ylim(0.,1.)
        plt.xticks(np.arange(0,1,0.1))
        plt.yticks(np.arange(0,1,0.1))
        plt.xticks(np.arange(0,1,0.1))
        plt.yticks(np.arange(0,1,0.1))
        plt.show()
def plt_(data):
    data = data.reshape(-1,3,60)
    for i in range(len(data)):
        plt.scatter(data[i][0][0:30],data[i][0][30:60],marker='*',color='red')
        plt.scatter(data[i][1][0:30],data[i][1][30:60],marker='*',color='blue')
        plt.scatter(data[i][2][0:30],data[i][2][30:60],marker='*',color='green')
        
        plt.plot(data[i][0][0:30],data[i][0][30:60],color='red')
        plt.plot(data[i][1][0:30],data[i][1][30:60],color='blue')
        plt.plot(data[i][2][0:30],data[i][2][30:60],color='green')
        plt.xlim(0.,1.)
        plt.ylim(0.,1.)
        plt.xticks(np.arange(0,1,0.1))
        plt.yticks(np.arange(0,1,0.1))
        plt.show()
plt_(data)