机器学习-数据可视化神器matplotlib学习之路(一)

直接上代码吧,说明写在备注就好了,这次主要学习一下基本的画图方法和常用的图例图标等

from matplotlib import pyplot as plt
import numpy as np

#这里是最最基本的代码了
#x轴-2到2均分50个点
x = np.linspace(-2, 2, 50)
y = x**2
plt.plot(x, y)
plt.show()

下面要加一下元素和样式了

from matplotlib import pyplot as plt
import numpy as np

x = np.linspace(-2, 2, 50)
y = x**2
plt.xlim((-3,3))#设置x坐标范围
plt.ylim((0,5))#设置y坐标范围
plt.plot(x, y,
         linestyle='--',#线条的样式
         linewidth=2.0,#线条宽度
         alpha=0.5,#透明度,数字越小越透明,范围(0-1)
         marker='.',#每个点的样式
         color='r',#颜色
         label='y=x**2'#图例
         )
#打开图例,不然plt.plot里面的label属性会无效
plt.legend(
    loc='upper right'#指定图例位置,这里是右上,默认为best会自动选择最优的位置
           )
plt.show()

到这里,如果我们想要x轴刻度自由控制一下可以用 plt.xticks 来实现

from matplotlib import pyplot as plt
import numpy as np

x = np.linspace(-2, 2, 50)
y = x**2
plt.xlim((-3,3))#设置x坐标范围
plt.ylim((0,5))#设置y坐标范围
plt.plot(x, y,
         linestyle='--',#线条的样式
         linewidth=2.0,#线条宽度
         alpha=0.5,#透明度,数字越小越透明,范围(0-1)
         marker='.',#每个点的样式
         color='r',#颜色
         label='y=x**2'#图例
         )
#打开图例,不然plt.plot里面的label属性会无效
plt.legend(
    loc='upper right'#指定图例位置,这里是右上,默认为best会自动选择最优的位置
           )

new_xticks = np.linspace(-2, 2, 6)
plt.xticks(new_xticks)#在-2到2之间平均显示6个刻度

plt.show()

这里x轴刻度就和之前发生一点变化,有时候不需要用具体数字而是一个文字标签显示,比如y轴0以上的为normal,2以上为good,4以上为very good,

可以用 plt.yticks([0, 2, 4], [‘normal’,'good','very good']) 来实现,接下来就来实现一下

from matplotlib import pyplot as plt
import numpy as np

x = np.linspace(-2, 2, 50)
y = x**2
plt.xlim((-3,3))#设置x坐标范围
plt.ylim((0,5))#设置y坐标范围
plt.plot(x, y,
         linestyle='--',#线条的样式
         linewidth=2.0,#线条宽度
         alpha=0.5,#透明度,数字越小越透明,范围(0-1)
         marker='.',#每个点的样式
         color='r',#颜色
         label='y=x**2'#图例
         )
#打开图例,不然plt.plot里面的label属性会无效
plt.legend(
    loc='upper right'#指定图例位置,这里是右上,默认为best会自动选择最优的位置
           )

new_xticks = np.linspace(-2, 2, 6)
plt.xticks(new_xticks)#在-2到2之间平均显示6个刻度

plt.yticks([0, 2, 4], [r'$normal$', r'$good$', r'$very\ good$'])#前后对应上就好了,$包住的是调整默认字体

plt.show()

接下来就是对坐标轴的处理了,我们想要移动坐标,用方框的左边和下班做y轴和x轴,右边和上班把颜色去掉,然后移动一下x和y轴

from matplotlib import pyplot as plt
import numpy as np

x = np.linspace(-2, 2, 50)
y = x**2
plt.xlim((-3,3))#设置x坐标范围
plt.ylim((0,5))#设置y坐标范围
plt.plot(x, y,
         linestyle='--',#线条的样式
         linewidth=2.0,#线条宽度
         alpha=0.5,#透明度,数字越小越透明,范围(0-1)
         marker='.',#每个点的样式
         color='r',#颜色
         label='y=x**2'#图例
         )
#打开图例,不然plt.plot里面的label属性会无效
plt.legend(
    loc='upper right'#指定图例位置,这里是右上,默认为best会自动选择最优的位置
           )

new_xticks = np.linspace(-2, 2, 6)
plt.xticks(new_xticks)#在-2到2之间平均显示6个刻度

plt.yticks([0, 2, 4], [r'$normal$', r'$good$', r'$very\ good$'])#前后对应上就好了,$包住的是调整默认字体

gca = plt.gca()
gca.spines['right'].set_color('none')#将图像方框右边颜色取消
gca.spines['top'].set_color('none')#将图像方框上边颜色取消
gca.xaxis.set_ticks_position('bottom')#x轴设为方框下边
gca.yaxis.set_ticks_position('left')#y轴设为方框左边
gca.spines['bottom'].set_position(('data', 1))#将方框下边也就是x轴移动到y=1的位置
gca.spines['left'].set_position(('data', 0))#将方框左边也就是y轴移动到x=0的位置

plt.show()

这里的方法稍微有点麻烦,折腾了一下终于大功告成了!今天就到这儿,下次准备学习一下更怎样在图中用文字和其它说明

posted @ 2019-01-10 17:49  依封剑白  阅读(395)  评论(0编辑  收藏  举报