Matplotlib 练习题

1. 绘制一个二维随机漫步的图形

直接上代码：

%pylab inline
nsteps = 1000
draws = np.random.randint(-1,2,size=(2,nsteps))
walks = draws.cumsum(1)
plot(walks[0,:],walks[1,:]);

Populating the interactive namespace from numpy and matplotlib

先生成 1000 个随机漫步方向，方向是从 {-1, 0, 1} 中随机挑两个值（两个值也可相等）作为移动方向，所以每次移动有 3×3=9 种选择，初始位置也是 9 种选择，cumsum 函数是将每次的移动累加，最后通过 plot 画出来。

2. 画出一个二次函数，同时画出梯形法求积分时的各个梯形

这里使用 IPython.html.widgets 模块中的 interact 函数，绘制一个交互式的函数图形。可以手动调整梯形个数，看到函数面积随梯形个数而变化。

def Quadratic(x): # 定义二次函数
    return 2*x**2 +3*x +4

import numpy as np
import matplotlib.pyplot as plt
from IPython.html.widgets import interact

def plot_ladder(laddernum):
    x = np.linspace(-5, 5, num=100)
    y = Quadratic(x)
    plot(x,y) # 先画出原函数的图形

    a = np.linspace(-5, 5, num=laddernum)
    for i in range(laddernum):
        plot([a[i],a[i]],[0,Quadratic(a[i])],color="black") # 画梯形的上底和下底

    ladders = [];
    for i in range(laddernum):
        ladders.append([a[i],Quadratic(a[i])]) # 因为梯形的腰是呈一条直线，所以这里存下各点坐标

    npladders = np.array(ladders)
    plot(npladders[:,0],npladders[:,1]); # 把梯形的斜腰连起来

interact(plot_ladder, laddernum=(1, 30, 1)) # 滑动模块在 1 和 30 之间变化，变化区间是 1

3. 研究 IPython.html.widgets 模块中的 interact 函数，绘制一个交互式的函数图形

2014 年 4 月，IPython 增加了 interactive widgets，提供了可以交互的界面组件，如下例：

from IPython.html.widgets import interact
%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
def plot_sine(n):
    x = np.arange(0, 20, 0.1)
    y = np.sin(x/n)
    plot(x,y)
interact(plot_sine, n=(1, 30, 0.1))

还可以这样用。

from IPython.html.widgets import interact, IntSlider # 把之前的画图函数改写成依赖于参数的函数

# 引入 interact 模块
from IPython.html.widgets import interact, IntSlider

# 把之前的画图函数改写成依赖于参数的函数
def test_interact(min_, max_, steps_):
    x = np.linspace(min_, max_, steps_)
    y = np.sin(x)
    z = np.cos(x)
    plot(x, y)
    scatter(x, y)

# 使用 interact 来交互式的调试参数
interact(test_interact,
         # 为每一个参数设定一个 interact 控件
         min_=IntSlider(min=1,        # 最小值
                              max=10,       # 最大值
                              step=1,       # 每次调节的步长
                              value=1),     # 初始值
         max_=IntSlider(min=10, max=20, step=1, value=10),
         steps_=IntSlider(min=10, max=100, step=10, value=50))

4. Seaborn

Matplotlib 是 Python 主要的绘图库，但它本身很复杂，它的图经过大量的调整才能变精致。因此，作为替代，推荐使用Seaborn。Seaborn本质上使用Matplotlib作为核心库（就像Pandas对NumPy一样）。它可以：

• 默认情况下就能创建赏心悦目的图表。
• 创建具有统计意义的图。
• 能理解pandas的DataFrame类型，所以它们一起可以很好地工作。

import seaborn as sns

# Load one of the data sets that come with seaborn
tips = sns.load_dataset("tips")

sns.jointplot("total_bill", "tip", tips, kind='reg');

参考资料

• 用 IPython notebook + nvd3 实现交互式调试
• Python 和数据科学的起步指南