Python 项目实践二（生成数据）第二篇

接着上节继续学习，在本节中，我们将使用Python来生成随机漫步数据，再使用matplotlib以引人瞩目的方式将这些数据呈现出来。随机漫步是这样行走得到的路径：每次行走都完全是随机的，没有明确的方向，结果是由一系列随机决策决定的。你可以这样认为，随机漫步就是蚂蚁在晕头转向的情况下，每次都沿随机的方向前行所经过的路径。

一 随机漫步

1 创建RandomWalk()类

为模拟随机漫步，我们将创建一个名为RandomWalk的类，它随机地选择前进方向。这个类需要三个属性，其中一个是存储随机漫步次数的变量，其他两个是列表，分别存储随机漫步经过的每个点的x和y坐标。代码如下：

from random import choice

class RandomWalk() :
    '''一个生成随机漫步数的类'''
    def __init__(self,num_points=50000):
        '''初始化随机漫步的属性'''
        self.num_points = num_points

        #所有随机漫步都始于（0,0）
        self.x_values = [0]
        self.y_values =[0]

2 选择方向

我们将使用fill_walk()来生成漫步包含的点，并决定每次漫步的方向，如下所示。请将这个方法添加到random_walk.py中：

 def fill_walk(self):
        #计算随机漫步包含的所有点

        #不断漫步，直到列表达到指定的长度
        while len(self.x_values) < self.num_points :

            #决定前进方向以及沿着个方向前进的距离
            x_direction = choice([1,-1])
            x_distance = choice([0,1,2,3,4])
            x_step = x_direction*x_distance

            y_direction = choice([1,-1])
            y_distance = choice([0,1,2,3,4])
            y_step = y_direction*y_distance

            #拒绝原地踏步
            if x_step == 0 and y_step ==0 :
                continue

            #计算下一个点的x和y值
            next_x= self.x_values[-1] + x_step
            next_y= self.y_values[-1] + y_step

            self.x_values.append(next_x)
            self.y_values.append(next_y)

代码注释的很清楚了，我再强调几点：

（1）我们建立了一个循环，这个循环不断运行，直到漫步包含所需数量的点。

（2）如何模拟四种漫步决定：向右走还是向左走？沿指定的方向走多远？向上走还是向下走？沿选定的方向走多远？

3 绘制随机漫步的图像

import matplotlib.pyplot as plt

from random_walk import RandomWalk

while True :
    random_walk = RandomWalk()
    random_walk.fill_walk()
    plt.figure(figsize=(10,6))
    point_numbers = list(range(random_walk.num_points))
    plt.scatter(random_walk.x_values,random_walk.y_values,c=point_numbers,cmap=plt.cm.Blues,edgecolor="none",s=1)

    # 突出起点和终点
    plt.scatter(0, 0, c='green', edgecolors='none', s=100)
    plt.scatter(random_walk.x_values[-1], random_walk.y_values[-1], c='red', edgecolors='none',
    s=100)

    #隐藏坐标轴
    plt.axes().get_xaxis().set_visible(False)
    plt.axes().get_yaxis().set_visible(False)
    plt.show()

    input_flag = input("是否结束？（Y/N）")

    if input_flag == "Y" :
        break

运行效果图如下：

注意几点：

（1）模拟多次随机漫步：每次随机漫步都不同，因此探索可能生成的各种模式很有趣。要在不多次运行程序的情况下使用前面的代码模拟多次随机漫步，一种办法是将这些代码放在一个while循环中。

（2）给点着色：我们将使用颜色映射来指出漫步中各点的先后顺序，并删除每个点的黑色轮廓，让它们的颜色更明显。为根据漫步中各点的先后顺序进行着色。

（3）重新绘制起点和终点：，可在绘制随机漫步图后重新绘制起点和终点。我们让起点和终点变得更大，并显示为不同的颜色，以突出它们。

（4）隐藏坐标轴：为修改坐标轴，使用了函数plt.axes()来将每条坐标轴的可见性都设置为False。随着你越来越多地进行数据可视化，经常会看到这种串接方法的方式。

（5）增加点数：增大num_points的值，并在绘图时调整每个点的大小。

（6）调整尺寸以适合屏幕：函数figure()用于指定图表的宽度、高度、分辨率和背景色。你需要给形参figsize指定一个元组，向matplotlib指出绘图窗口的尺寸，单位为英寸。

二 使用Pygal模拟掷骰子

在本节中，我们将使用Python可视化包Pygal来生成可缩放的矢量图形文件。对于需要在尺寸不同的屏幕上显示的图表，这很有用，因为它们将自动缩放，以适合观看者的屏幕。如果你打算以在线方式使用图表，请考虑使用Pygal来生成它们，这样它们在任何设备上显示时都会很美观。

在这个项目中，我们将对掷骰子的结果进行分析。掷6面的常规骰子时，可能出现的结果为1~6点，且出现每种结果的可能性相同。然而，如果同时掷两个骰子，某些点数出现的可能性将比其他点数大。为确定哪些点数出现的可能性最大，我们将生成一个表示掷骰子结果的数据集，并根据结果绘制出一个图形。

1 安装Pygal

python -m pip install --user pygal

2 创建Die类

from random import randint

class Die():
    '''表示一个骰子的类'''

    def __init__(self,num_sides=6):
        '''骰子默认6个面'''
        self.num_sides=num_sides

    def roll(self):
        #返回一个1-面数之间的随机值
        return randint(1,self.num_sides)

3 掷筛子

使用这个类来创建图表前，先来掷D6骰子，将结果打印出来，并检查结果是否合理：

from die import Die

die=Die()

results=[]

for roll_num in range(100) :
    result=die.roll()
    results.append(result)

print(results)

结果如下：

4 分析结果

为分析掷一个D6骰子的结果，我们计算每个点数出现的次数：

from die import Die

die=Die()

results=[]

for roll_num in range(1000) :
    result=die.roll()
    results.append(result)

#分析结果
frequencies=[]
for value in range(1,die.num_sides+1) :
    frequency=results.count(value)
    frequencies.append(frequency)

print(frequencies)

结果如下：[156, 171, 151, 173, 181, 168]

结果看起来是合理的：我们看到了6个值——掷D6骰子时可能出现的每个点数对应一个；我们还发现，没有任何点数出现的频率比其他点数高很多。下面来可视化这些结果。

5 绘制立方图

from die import Die
import pygal

die=Die()

results=[]

for roll_num in range(1000) :
    result=die.roll()
    results.append(result)

#分析结果
frequencies=[]
for value in range(1,die.num_sides+1) :
    frequency=results.count(value)
    frequencies.append(frequency)

#print(frequencies)

#对结果进行可视化
hist = pygal.Bar()

hist.title= "Results of rolling one D6 1000 times."
hist.x_labels=['1','2','3','4','5','6']
hist.x_title = "Result"
hist.y_title = "Frequency of Result"

hist.add("D6",frequencies)
hist.render_to_file("die_visual.svg")

为创建条形图，我们创建了一个pygal.Bar()实例，并将其存储在hist中,我们设置hist的属性title（用于标示直方图的字符串），将掷D6骰子的可能结果用作x轴的标签，并给每个轴都添加了标题,我们使用add()将一系列值添加到图表中（向它传递要给添加的值指定的标签，还有一个列表，其中包含将出现在图表中的值）。最后，我们将这个图表渲染为一个SVG文件，这种文件的扩展名必须为.svg。
要查看生成的直方图，最简单的方式是使用Web浏览器。如下图：

6 同时掷两个骰子

每次掷两个骰子时，我们都将两个骰子的点数相加，并将结果存储在results中。请复制die_visual.py并将其保存为dice_visual.py，再做如下修改：

from die import Die
import pygal

die_1=Die()
die_2=Die()

results=[]

for roll_num in range(1000) :
    result=die_1.roll() + die_2.roll()
    results.append(result)

#分析结果
frequencies=[]
max_result = die_1.num_sides + die_2.num_sides

for value in range(2,max_result+1) :
    frequency=results.count(value)
    frequencies.append(frequency)

#print(frequencies)

#对结果进行可视化
hist = pygal.Bar()

hist.title= "Results of rolling two D6 1000 times."
hist.x_labels=['1','2','3','4','5','6''7','8','9','10','11','12']
hist.x_title = "Result"
hist.y_title = "Frequency of Result"

hist.add("D6+D6",frequencies)
hist.render_to_file("die_visual.svg")

结果如下：

也可以掷不同的骰子，就不在详解了！未完待续！