Python数据可视化(二) 随机漫步

随机漫步:每次行走都是完全随机的,没有明确的方向,结果由一系列随机决策决定。

创建RandomWalk()类

创建一个名为RandomWalk的类,它随机地选择前进方向。这个类需要三个属性,一个存储随机漫步次数,其他两个是列表,分别存储随机漫步经过的每个点的x和y坐标。同时,RandomWalk类包含两个方法:init()和fill_walk(),后者用于计算随机漫步经过的所有点。

from random import choice 

class RandomWalk(): 
    """一个生成随机漫步数据的类""" 
    def __init__(self, num_points=5000): 
        """初始化随机漫步的属性""" 
        self.num_points = num_points 
        # 所有随机漫步都始于(0,0) 
        self.x_values = [0] 
        self.y_values = [0]

选择方向

使用fill_walk()来生成漫步包含的点,并决定每次漫步的方向。

    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)

使用choice([1,-1])给x_direction选择一个值,结果要么是表示向右走的1,要么是表示向左走的-1。接下来,choice([0, 1, 2, 3, 4])随机地选择一个0~4之间的整数,告诉Python沿指定的方向走多远(x_distance)。为获取漫步中下一个点的x值,将x_step与x_values中的最后一个值相加,对y值也做相同的处理。获得下一个点的x值和y值后,将它们分别附加到列表x_values和y_values的末尾。

绘制随机漫步图

新建一个.py文件,绘制随机漫步的所有点:

import matplotlib.pyplot as plt 
from random_walk import RandomWalk 


# 创建一个RandomWalk实例,并将其包含的点都绘制出来 
rw = RandomWalk() 
rw.fill_walk()
plt.scatter(rw.x_values, rw.y_values, s=15) 
plt.show()

这里导入的RandomWalk类是前述两段代码的总和,它们应该保存在另一个.py文件中,且与当前的程序文件同目录。


模拟多次随机漫步

添加一个while循环,在matplotlib查看器中显示结果,再在不关闭查看器的情况下暂停。如果关闭查看器,程序将询问是否要再模拟一次随机漫步。如果输入y,可模拟多次随机漫步;要结束程序,请输入n

import matplotlib.pyplot as plt 
from random_walk import RandomWalk 

while True:
    # 创建一个RandomWalk实例,并将其包含的点都绘制出来 
    rw = RandomWalk() 
    rw.fill_walk()
    plt.scatter(rw.x_values, rw.y_values, s=15) 
    plt.show()

    keep_running = input("Make another walk? (y/n): ")
    if keep_running == 'n': 
        break

给点着色

使用颜色映射来指出漫步中各点的先后顺序,并删除每个点的黑色轮廓。为此,在调用scatter()时传递参数c,并将其设置为一个列表,其中包含各点的先后顺序,最终的随机漫步图从浅蓝色渐变为深蓝色,代码如下:

--snip--
while True: 
    # 创建一个RandomWalk实例,并将其包含的点都绘制出来 
    rw = RandomWalk() 
    rw.fill_walk()
    point_numbers = list(range(rw.num_points)) 
    plt.scatter(rw.x_values, rw.y_values, c=point_numbers, cmap='Blues', edgecolor='none', s=15)
    plt.show() 

    keep_running = input("Make another walk? (y/n): ")
     --snip--

重新绘制起点和终点,并隐藏坐标轴

绘制随机漫步图后重新绘制起点和终点,让起点和终点变得更大,并显示为不同的颜色,以突出它们。使用函数plt.axes()将每条坐标轴的可见性都设置为False。

--snip--
while True:
     --snip--
     plt.scatter(rw.x_values, rw.y_values, c=point_numbers, cmap='Blues', edgecolor ='none', s=15) 

     # 突出起点和终点 
     plt.scatter(0, 0, c='green', edgecolors='none', s=100)
     plt.scatter(rw.x_values[-1], rw.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()
     --snip--

调整尺寸以适合屏幕

--snip--
while True: 
# 创建一个RandomWalk实例,并将其包含的点都绘制出来 
     rw = RandomWalk()
     rw.fill_walk() 
     # 设置绘图窗口的尺寸 
     plt.figure(figsize=(10, 6))
     --snip--

函数figure()用于指定图表的宽度、高度、分辨率和背景色,需要给形参figsize指定一个元组,向matplotlib指出绘图窗口的尺寸,单位为英寸。Python假定屏幕分辨率为80像素/英寸,如果你知道自己系统的分辨率,可使用形参dpi向figure()传递该分辨率,以有效地利用可用的屏幕空间:
plt.figure(dpi=128, figsize=(10, 6))

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,192评论 6 511
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,858评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,517评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,148评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,162评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,905评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,537评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,439评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,956评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,083评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,218评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,899评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,565评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,093评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,201评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,539评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,215评论 2 358

推荐阅读更多精彩内容