当前位置 博文首页 > python绘制高斯曲线

    python绘制高斯曲线

    作者:一个超会写Bug的安太 时间:2021-07-26 17:43

    本文实例为大家分享了python绘制高斯曲线的具体代码,供大家参考,具体内容如下

    源码:

    import numpy as np
    import matplotlib.pyplot as plt
    import math
    import mpl_toolkits.mplot3d
    import tensorflow.compat.v1 as tf
    tf.disable_v2_behavior()
    
    #import tensorflow as tf
    from sklearn import datasets
    
    sess = tf.InteractiveSession()
    gamma = tf.constant(-1.0)
    x, y = np.mgrid[-2:2:0.01, -2:2:0.01]
    
    x_data = tf.placeholder(shape=[400, 400], dtype=tf.float32)
    y_data = tf.placeholder(shape=[400, 400], dtype=tf.float32)
    
    Kernel = tf.exp(tf.multiply(gamma, tf.add((x_data*x_data),(y_data*y_data))))
    Kernel = sess.run(Kernel, feed_dict={x_data: x,y_data: y})
    
    ax = plt.subplot(111, projection='3d')
    ax.plot_surface(x, y, Kernel, rstride=1, cstride=1, cmap='rainbow', alpha=0.9)#绘面
    ax.set_xlabel('x')
    ax.set_ylabel('y')
    ax.set_zlabel('Kernel')
    plt.show()

    效果图:

    jsjbwy