INT104-lab9 测试新数据集

import numpy as np
from sklearn.manifold import TSNE
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D


def read(path: str) -> list:
    with open(path, "r") as f:
        text = f.readlines()
        D = []
        for row in text:
            row = str.replace(row, "14:", ",")
            row = str.replace(row, "13:", ",")
            row = str.replace(row, "12:", ",")
            row = str.replace(row, "11:", ",")
            row = str.replace(row, "10:", ",")
            row = str.replace(row, "9:", ",")
            row = str.replace(row, "8:", ",")
            row = str.replace(row, "7:", ",")
            row = str.replace(row, "6:", ",")
            row = str.replace(row, "5:", ",")
            row = str.replace(row, "4:", ",")
            row = str.replace(row, "3:", ",")
            row = str.replace(row, "2:", ",")
            row = str.replace(row, "1:", ",")
            row = str.replace(row, ",-1 1,", ",")
            row = str.replace(row, ",1 1,", ",")
            substr = str.split(str.split(row, "
")[0], ",")
            X = []
            for a in substr:
                X.append(float(a))
            D.append(X)
    return D


def init(D: list) -> tuple:
    n, m = len(D), len(D[0])
    X, Y = [], []
    for i in range(n):
        x = []
        if len(D[i]) != m:
            n -= 1
            continue
        for j in range(1, m, 1):
            x.append(D[i][j])
        X.append(x)
        Y.append(D[i][0])
    # print(n, m)
    return X, Y, n, m - 1


if __name__ == '__main__':

    D = read("binary_data.txt")
    X, Y, n, m = init(D)

    x = np.array(X)

    tsne = TSNE(n_components=3)

    tsne.fit_transform(x)

    one_x, one_y, one_z, zero_x, zero_y, zero_z = [], [], [], [], [], []
    for i in range(n):
        _x, _y, _z = tsne.embedding_[i][0], tsne.embedding_[i][1], tsne.embedding_[i][2]
        if Y[i] > 0:
            zero_x.append(_x)
            zero_y.append(_y)
            zero_z.append(_z)
        else:
            one_x.append(_x)
            one_y.append(_y)
            one_z.append(_z)
    '''
    ax = plt.axes(projection='3d')
    ax.scatter3D(one_x, one_y, one_z)
    ax.scatter3D(zero_x, zero_y, zero_z)
    '''
    plt.subplot(311)
    plt.scatter(one_x, one_y)
    plt.scatter(zero_x, zero_y)
    plt.subplot(312)
    plt.scatter(one_x, one_z)
    plt.scatter(zero_x, zero_z)
    plt.subplot(313)
    plt.scatter(one_y, one_z)
    plt.scatter(zero_y, zero_z)
    plt.show()