之前有篇文章，说房间里有100个人，每人100块钱，的那个
原文介绍：用数据分析告诉你这个世界很有意思
觉得挺有意思的，昨天发现pyplot也可以绘制动画，就来试试，主要是实现动画效果，其他的暂时先不考虑了
目前跑起来是可以，就是比较慢，还没找到原因
整体想法，就是
x轴表示玩家的序号，财富值用y轴来表示
存款为0后，不在支付给别人，但可以收到别人给的钱
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个人那么高
代码跑的很慢，随机那块不知道有没有问题，等再研究下看看优化下
可以贷款，即存款为负数
#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就可以解决，一会儿找时间改下看看
后续再补充下其他情况
网友评论