第五章：matplotlib水印和桑基图

1、Matplotlib水印

 1 import matplotlib.pyplot as plt
 2 import numpy as np
 3
 4 x = np.linspace(0.0,10,40)
 5 y = np.random.randn(40)
 6
 7 plt.plot(x,                 # X轴数据
 8          y,                 # Y轴数据
 9          ls="-",            # 线条样式
10          lw=2,              # 线条宽度
11          marker="o",        # x轴Y轴对应数据点的样式
12          ms=20,             # 数据点的大小
13          mfc="orange",      # 数据点填充颜色
14          alpha=0.6)         # 数据点的透明度
15 plt.grid(ls=":",            # 网格样式（虚线）
16          color="gray",      # 网格颜色
17          alpha=0.5)         # 网格透明度
18 plt.text(x=1,               # 水印开头左下角对应的X点
19          y=2,               # 水印开头左下角对应的Y点
20          s="Matplotlib",    # 水印文本
21          fontsize=50,       # 水印大小
22          color="gray",      # 水印颜色
23          alpha=0.5)         # 水印是通过透明度控制的
24 plt.show()

2、桑基图

 1 import  matplotlib
 2 import  matplotlib.pyplot as plt
 3 from matplotlib.sankey import Sankey
 4
 5 # 显示中文标识
 6 matplotlib.rcParams["font.sans-serif"] = ["SimHei"]
 7 matplotlib.rcParams["axes.unicode_minus"] = False
 8
 9 # 消费收入与支出数据
10 flows = [0.2,0.1,0.4,0.3,-0.6,-0.05,-0.15,-0.2]
11
12 # 数据标签来源
13 labels = ["卖白菜","卖萝卜","卖肉","卖烟","买鞋","买衣服","买苹果","买书"]
14
15 # 1表示流量显示在上方，0表示流量显示在水平位置，-1表示流量显示在下方
16 orientations = [1,1,0,-1,1,-1,1,0]
17
18 # 通过Sankey创建一个基本的桑基图对象
19 sankey = Sankey()
20
21 # 为桑基图添加绘制数据
22 sankey.add(flows=flows,                 # 收入支出数据
23            labels=labels,               # 数据标签
24            orientations=orientations,   # 每个数据显示在上方或者下方或者水平
25            color="r",                   # 桑基图边缘线条颜色
26            fc="lightgreen",             # 桑基图填充颜色
27            patchlabel="生活消费",        # 桑基图标题
28            alpha=0.7)                   # 桑基图透明度
29
30 # 桑基图基本绘制完成对象
31 diagrams = sankey.finish()
32 diagrams[0].texts[4].set_color("r")     # 数据标签来源下标为4设置为红色
33 diagrams[0].texts[4].set_weight("bold") # 数据标签来源下标为4字体设置加粗
34 diagrams[0].text.set_fontsize(20)       # 生活消费的字号大小
35 diagrams[0].text.set_fontweight("bold") # 生活消费字体加粗
36
37 plt.title("日常生活中的开支流量图")
38 plt.show()