‘‘‘
-- coding: utf-8 --
Created on Fri Oct 12 12:31:21 2018
項(xiàng)目 13 社會(huì)財(cái)富分配問(wèn)題 （蒙特卡羅模擬）
Note:
1 建立一個(gè)空的DataFrame時(shí)薇溃，只需要index 參數(shù)
2 當(dāng)在一個(gè)列表中隨機(jī)選取一個(gè)值 可以用random的choice纺腊，當(dāng)需要隨機(jī)選取多個(gè)值用numpy
的random的choice()
3 Series 的name 參數(shù)設(shè)置在Series 中 name=‘’
4 當(dāng)在DataFrame需要判斷再賦值時(shí)馆揉，可以先用判斷篩選列 重新賦值 或者 用apply函數(shù)
5 apply在DataFrame中用于判斷賦值（☆）
6 在繪制圖表時(shí) 如果不想顯示每一個(gè)xticks 不要xlim的?
7 在迭代過(guò)程中不要將值賦值給變量名和傳入的變量一致漾唉，會(huì)導(dǎo)致報(bào)錯(cuò)
8 在篩選時(shí)或者標(biāo)記 當(dāng)對(duì)某些值進(jìn)行篩選或者按某些條件篩選 即按照此標(biāo)記
"""
導(dǎo)入模塊
import os
import random
import time
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import warnings
warnings.filterwarnings('ignore')
定義函數(shù)
def round1():
'''
建立初始模型灰殴，第1次交易
'''
# 構(gòu)建初始財(cái)富值100涩搓，index的值為每個(gè)人的編號(hào)
people = pd.DataFrame(index=list(range(1,101)))
people['money'] = 100
people['r1'] = people['money'] - 1
# 構(gòu)建收錢(qián)的隨機(jī)對(duì)象
people['to'] = np.random.choice(list(range(1,101)), size=100,
replace=True, p=None)
data_to = people['to'].value_counts()
data_to.name = 'count'
data_to = pd.DataFrame(data_to)
people = pd.merge(people, data_to, how='left', left_index=True,
right_index=True).fillna(0)
people['r1_m'] = people['r1'] + people['count']
return people
def roundi(n):
'''
構(gòu)建借貸模型
不考慮財(cái)富值為0蛙紫，即允許借貸
'''
# 構(gòu)建初始財(cái)富值
people = pd.DataFrame(index=list(range(1,101)))
people['r0'] = 100
# 構(gòu)建模型
for i in range(1, n+1):
col = 'r' + str(i)
col0 = 'r' + str(i-1)
people[col] = people[col0] - 1
data_to = pd.Series(np.random.choice(list(range(1,101)), size=100,
replace=True, p=None), name='to')
data_to = data_to.value_counts()
data_to = pd.DataFrame(data_to)
people = pd.merge(people, data_to, how='left', left_index=True,
right_index=True).fillna(0)
people[col] = people[col] + people['to']
del people['to']
return people.T
def roundn(n):
'''
構(gòu)建初始模型
考慮財(cái)富值為0拇派，即財(cái)富值為0時(shí)可以接收 ==> choice的范圍是1-100
但不給出 ==> 所以要統(tǒng)計(jì)給出的數(shù)量 == 接收的數(shù)量
'''
# 構(gòu)建初始財(cái)富值
people = pd.DataFrame(index=list(range(1,101)))
people['r0'] = 100
# 構(gòu)建模型
for i in range(1, n+1):
col = 'r' + str(i)
col0 = 'r' + str(i-1)
people[col] = people[col0] - 1
data_to = pd.Series(np.random.choice(list(range(1,101)), size=100,
replace=True, p=[]), name='to')
data_to = data_to.value_counts()
data_to = pd.DataFrame(data_to)
people = pd.merge(people, data_to, how='left', left_index=True,
right_index=True).fillna(0)
people[col] = people[col] + people['to']
del people['to']
return people.T
def roundm(n):
'''
構(gòu)建努力人生模型
'''
# 構(gòu)建初始財(cái)富值
people = pd.DataFrame(index=list(range(1,101)))
people['r0'] = 100
person_id= [1, 11, 21, 31, 41, 51, 61, 71, 81, 91] # 努力的Id
# 構(gòu)建概率
p = [0.899/90 for i in range(100)]
for i in person_id:
p[i-1] = 0.0101
# 構(gòu)建模型
for i in range(1, n+1):
col = 'r' + str(i)
col0 = 'r' + str(i-1)
people[col] = people[col0] - 1
data_to = pd.Series(np.random.choice(list(range(1,101)), size=100,
replace=True, p=p), name='to')
data_to = data_to.value_counts()
data_to = pd.DataFrame(data_to)
people = pd.merge(people, data_to, how='left', left_index=True,
right_index=True).fillna(0)
people[col] = people[col] + people['to']
del people['to']
return people.T
def graph(data, title):
'''
繪制柱狀圖 == 表排序
'''
plt.figure(figsize=(10, 5))
data.plot(kind='bar', color='gray', edgecolor='gray', figsize=(10, 5))
plt.xlabel('Player Id')
plt.ylabel('Forturn')
plt.title(title)
plt.savefig(title + '.jpg', dpi=200)
def lst():
'''
生產(chǎn)繪制圖表數(shù)據(jù)的行數(shù)
'''
lst1 = [x for x in range(0, 100, 10)]
lst2 = [x for x in range(100, 1000, 100)]
lst3 = [x for x in range(1000, 17001, 400)]
return lst1 + lst2 + lst3
def forturn_std(data):
'''
計(jì)算每一輪的財(cái)富標(biāo)準(zhǔn)差
'''
lst = []
for i in range(17001):
dat = data.iloc[i]
std = dat.std()
lst.append(std)
s = pd.Series(lst)
return s
def line_graph(data, title):
'''
繪制折線圖
'''
fig = plt.figure(figsize=(10,5))
plt.plot(data, color='red')
plt.grid(linestyle='--', color='gray', alpha=0.6, axis='both')
plt.xlim([0, 17000])
plt.ylim([0, 150])
plt.title(title)
plt.savefig(title + '.jpg', dpi=400)
def sign_pc(data):
'''
負(fù)債id標(biāo)記
'''
data = pd.DataFrame(data)
data['color'] = 'gray'
data['color'][data['r6200'] < 0] = 'red'
del data['r6200']
return data
if name == 'main':
# 運(yùn)行初始模型 得到模型數(shù)據(jù)
r17 = roundn(17000)
# 繪制圖表 -- 不排序
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表\初始不排序'
os.chdir(path)
lst = lst()
for i in lst:
title = 'Round' + str(i)
#data = r17.iloc[i]
#graph(data, title)
# 繪制圖表 -- 排序
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表\初始排序'
os.chdir(path)
for i in lst:
title = 'Round' + str(i)
data = r17.iloc[i]
data = data.sort_values(ascending=True)
graph(data, title)
# 運(yùn)行借貸模型荷辕，得到數(shù)據(jù)
ri17 = roundi(17000)
# 繪制圖表
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表\允許借貸'
os.chdir(path)
for i in lst:
title = 'Round' + str(i)
data = ri17.iloc[i]
data = data.sort_values(ascending=True)
graph(data, title)
# 調(diào)用計(jì)算函數(shù)，獲取標(biāo)準(zhǔn)差
data_std = forturn_std(ri17)
# 繪制圖表
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表'
os.chdir(path)
line_graph(data_std, '財(cái)富標(biāo)準(zhǔn)差曲線')
# 35歲破產(chǎn)往后逆襲情況 6200次
data_6200 = ri17.iloc[6200]
id_pc =data_6200[data_6200 < 0].index.tolist()
# 對(duì)負(fù)債id進(jìn)行標(biāo)記
id_sign = sign_pc(data_6200)
# 繪制圖表
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表\負(fù)債逆襲'
os.chdir(path)
for i in range(6200, 17000, 500):
title = 'Round' + str(i)
col = 'r' + str(i)
data = ri17.iloc[i]
data = pd.DataFrame(data)
data1 = pd.merge(data, id_sign, how='right', left_index=True, right_index=True)
data1 = data1.sort_values(by=col, ascending=True)
fig = plt.figure(figsize=(10,5))
data1[col].plot(kind='bar', color=data1['color'], figsize=(10,5))
plt.xlabel('Player Id')
plt.ylabel('Forturn')
plt.title(title)
plt.savefig(title + '.jpg', dpi=200)
# 運(yùn)行努力人生模型件豌，得到數(shù)據(jù)
rm17 = roundm(17000)
# 對(duì)努力id進(jìn)行標(biāo)記
id_nl = [1, 11, 21, 31, 41, 51, 61, 71, 81, 91]
id_nl_sign = pd.Series('gray', index=range(1, 101), name='color')
for i in id_nl:
id_nl_sign[i] = 'red'
id_nl_sign = pd.DataFrame(id_nl_sign)
# 繪制圖表
path = r'C:\Users\pj2063150\Desktop\項(xiàng)目\項(xiàng)目13社會(huì)財(cái)富分配問(wèn)題模擬\圖表\努力人生'
os.chdir(path)
for i in lst:
title = 'Round' + str(i)
col = 'r' + str(i)
data = rm17.iloc[i]
data = pd.DataFrame(data)
data = pd.merge(data, id_nl_sign, how='right', left_index=True, right_index=True)
data = data.sort_values(by=col, ascending=True)
fig = plt.figure(figsize=(10,5))
data[col].plot(kind='bar', color=data['color'], figsize=(10,5))
plt.xlabel('Player Id')
plt.ylabel('Forturn')
plt.title(title)
plt.savefig(title + '.jpg', dpi=200)
print('Finished')
’’’
