Представьте данные ncoll и wcoll как 4000-ю случайные числа.
Я хочу запустить их через TSNE, а затем создать трехмерный график.
Если я построю это, у меня получится двумерный график, поэтому что-то идет не так, но я не совсем уверен, что.
В конечном итоге я хочу изобразить первую половину красным, а вторую синим - на одном и том же трехмерном графике.
print(__doc__)
from time import time
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib.ticker import NullFormatter
from sklearn import manifold
from sklearn.utils import check_random_state
c = 1000
# Open File
ncoll_fn = "C:/Users/xxlassi/Downloads/trajectory_demo/trajectory_270252769939974_run__uid_-1808183947_tag_collision_0.0.txt"
wcoll_fn = "C:/Users/xxlassi/Downloads/trajectory_demo/trajectory_271551342150600_run__uid_-918721219_tag_collision_0.01.txt"
ncoll = []
wcoll = []
with open( ncoll_fn ) as f:
ncoll = [ np.array([ float(el) for el in line.strip().split(',') ]) for line in f.readlines() ]
ncoll = np.array( ncoll )
with open( wcoll_fn ) as f:
wcoll = [ np.array([ float(el) for el in line.strip().split(',') ]) for line in f.readlines() ]
wcoll = np.array( wcoll )
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
arr = np.concatenate((wcoll, ncoll), axis=0)
mid = int(len(arr)/2)
print (mid)
tsne = manifold.TSNE(n_components=3, init='pca',random_state=0, perplexity= 30, n_iter=5000)
trans_data = tsne.fit_transform(arr)
ax.scatter(arr[:,0][0:mid], arr[:,1][0:mid], c= 'r')
ax.scatter(arr[:,0][mid:], arr[:,1][mid:], c= 'b')
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
plt.title("t-SNE")
plt.axis('tight')
plt.show()