1. 程式人生 > >python 多維高斯分佈資料生成

python 多維高斯分佈資料生成


import numpy as np
import matplotlib.pyplot as plt


def gen_clusters():
    mean1 = [0,0]
    cov1 = [[1,0],[0,10]]
    data = np.random.multivariate_normal(mean1,cov1,100)
    
    mean2 = [10,10]
    cov2 = [[10,0],[0,1]]
    data = np.append(data,
                     np.random.multivariate_normal(mean2,
cov2,100), 0) mean3 = [10,0] cov3 = [[3,0],[0,4]] data = np.append(data, np.random.multivariate_normal(mean3,cov3,100), 0) return np.round(data,4) def save_data(data,filename): with open(filename,'w') as file: for
i in range(data.shape[0]): file.write(str(data[i,0])+','+str(data[i,1])+'\n') def load_data(filename): data = [] with open(filename,'r') as file: for line in file.readlines(): data.append([ float(i) for i in line.split(',')]) return np.array(data)
def show_scatter(data): x,y = data.T plt.scatter(x,y) plt.axis() plt.title("scatter") plt.xlabel("x") plt.ylabel("y") data = gen_clusters() save_data(data,'3clusters.txt') d = load_data('3clusters.txt') show_scatter(d)

在這裡插入圖片描述