proj.py 文件源码

def plot_reg_2D_stoc(X,stoc_vector):

    deter_vec = np.invert(stoc_vector)

    dom_max = np.amax(X[stoc_vector,:]) + 1

    A = np.zeros((dom_max,dom_max))

    stoc_indexs = np.arange(0,X.shape[0],1)[stoc_vector].astype(int)

    for i,deter_element in enumerate(deter_vec):
        if deter_element == True:
            A[X[int(stoc_indexs[0]),:].astype(int), X[int(stoc_indexs[1]),:].astype(int)] = X[i,:]
            pl.figure(i)
            #ax = fig.gca(projection='3d')
            #surf = ax.plot_surface(X[int(stoc_indexs[0]),:].astype(int), X[int(stoc_indexs[1]),:].astype(int),X[i,:], rstride=1, cstride=1,
#cmap=cm.coolwarm,linewidth=0, antialiased=False)
            pl.contour(A,X[i,:])
            #ax.zaxis.set_major_locator(LinearLocator(10))
            #ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))
            #fig.colorbar(surf, shrink=0.5, aspect=5)
            pl.show()