matplotlib手册(10)-用pyplot实现“房间里100个人玩游戏的例子”

发表于 | 分类于 | | 本文总阅读量 次

之前有篇文章，说房间里有100个人，每人100块钱，的那个
原文介绍：用数据分析告诉你这个世界很有意思

觉得挺有意思的，昨天发现pyplot也可以绘制动画，就来试试，主要是实现动画效果，其他的暂时先不考虑了
目前跑起来是可以，就是比较慢，还没找到原因
整体想法，就是
x轴表示玩家的序号，财富值用y轴来表示

存款为0后，不在支付给别人，但可以收到别人给的钱

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import numpy as np  
import matplotlib.pyplot as plt  
import matplotlib.animation as animation  
#初始状态，共100人
total_people = 100  
#每个人的x轴坐标
x = np.arange(100)
#每人初始100块
people = [100]*total_people
#局数
game_round = 0
game_max_round = 17000
#初始绘图
fig,axes = plt.subplots()
axes.bar(x,people,facecolor='green') 
axes.set_title(u'Round: '+str(game_round))
axes.grid(True,axis='y')
#根据下标i,随机返回另一个下标
def give_to(i):
    i_to = i
    while i == i_to:
        i_to = np.random.randint(0,100)
    
    #print('from',i,'to',i_to)
    return i_to
#重新绘制图形
#1.当拥有的钱为0，则不再支出，但可以收入
def game(obj): 
    global people
    global game_round
    #还不知道咋让循环停止，就在这判断下
    if game_round < game_max_round:
        #清空当前轴
        plt.cla()
        
        #遍历100个人
        for i in range(total_people):
            #判断，当前人是否有钱
            if people[i] > 0 :
                #每个人拿出1块钱，给另一个人
                people[i] = people[i] - 1
                people_to = give_to(i)
                people[people_to] = people[people_to] + 1
                
                #print(people)
            else :
                pass
         
        
        game_round += 1
        #重新绘图
        axes.set_title(u'Round: '+str(game_round))
        axes.bar(x,sorted(people),facecolor='green')
        axes.grid(True,axis='y')
    else :
        pass
#循环调用游戏
ani = animation.FuncAnimation(fig, game, interval=0.1)  
plt.show()

不知道是不是代码有问题，最后有1个人那么高
代码跑的很慢，随机那块不知道有没有问题，等再研究下看看优化下

可以贷款，即存款为负数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#2.允许借贷的情况，及拥有的钱可以为负
def game2(obj): 
    global people
    global game_round
    #还不知道咋让循环停止，就在这判断下
    if game_round < game_max_round:
        #清空当前轴
        plt.cla()
        
        #遍历100个人
        for i in range(total_people):
            #每个人拿出1块钱，给另一个人
            people[i] = people[i] - 1
            people_to = give_to(i)
            people[people_to] = people[people_to] + 1
         
        
        game_round += 1
        #重新绘图
        axes.set_title(u'Round: '+str(game_round))
        axes.bar(x,sorted(people),facecolor='green')
        axes.grid(True,axis='y')
    else :
        pass

35岁破产后，人生的走势

这里就以第6500轮游戏为基准，

# -*- coding: utf-8 -*-
"""
Created on Fri Aug 18 11:08:44 2017

@author: yuguiyang
"""
import numpy as np  
import matplotlib.pyplot as plt  
import matplotlib.animation as animation

#定义一个people类
class People:
    __data = 100
    __color = 'green'


    def __init__(self,data=100,color='green'):
        self.__data = data
        self.__color = color

    def __str__(self):
        return str(self.__data)+','+self.__color

    def set_data(self,data):
        self.__data = data

    def set_color(self,color):
        self.__color = color

    def get_data(self):
        return self.__data

    def get_color(self):
        return self.__color

    def give_money(self,money=1):
        self.__data = self.__data - money

    def rcv_money(self,money=1):
        self.__data = self.__data + money


#对数据进行排序后返回数据
def parse_people_data():
    global peoples

    people_data = []
    people_color = []
    for p in sorted(peoples, key=lambda p: p.get_data()):
        people_data.append(p.get_data())
        people_color.append(p.get_color())


    return people_data,people_color


####################################################

#初始状态，共100人
total_people = 100  
#每个人的x轴坐标
x = np.arange(total_people)

#每人初始100块,颜色是绿色
peoples = []
for i in range(total_people):
    peoples.append(People())


#局数
game_round = 0
game_max_round = 17000

#初始绘图
fig,axes = plt.subplots()

axes.bar(x,parse_people_data()[0],color=parse_people_data()[1]) 
axes.set_title(u'Round: '+str(game_round))
axes.grid(True,axis='y')


#根据下标i,随机返回另一个下标
def give_to(i):
    i_to = i
    while i == i_to:
        i_to = np.random.randint(0,100)

    return i_to

#重新绘制图形
#1.当拥有的钱为0，则不再支出，但可以收入
#参考plt_flash_demo2

#2.允许借贷的情况，及拥有的钱可以为负
def game2(obj): 
    global peoples
    global game_round

    #还不知道咋让循环停止，就在这判断下
    if game_round < game_max_round:
        #清空当前轴
        plt.cla()

        #遍历100个人
        for i in range(total_people):
            #每个人拿出1块钱，给另一个人
            peoples[i].give_money()
            people_to = give_to(i)
            peoples[people_to].rcv_money()

        #在第6500次游戏，修改负债者的颜色
        if game_round == 6500:
            for p in peoples :
                if p.get_data()<0:
                    p.set_color('red')
                else :
                    pass

        game_round += 1
        #重新绘图
        axes.set_title(u'Round: '+str(game_round))
        axes.bar(x,parse_people_data()[0],color=parse_people_data()[1])
        axes.grid(True,axis='y')
    else :
        pass
#循环调用游戏
ani = animation.FuncAnimation(fig, game2, interval=1)  

plt.show()

6500忘记截图了，

挺好玩儿的，还是有2个人逆袭的
这里，定义了一个People类来绑定财富值和颜色，感觉可以直接用pandas DataFrame就可以解决，一会儿找时间改下看看
后续再补充下其他情况

于贵洋 wechat
要教我弹吉他嘛！