# die.py 骰子類模塊
from random import randint
class Die():
  """骰子類"""
  def __init__(self, num_sides=6):
    """默認(rèn)六面的骰子"""
    self.num_sides = num_sides
  def roll(self):
    """擲骰子的方法"""
    return randint(1, self.num_sides)

# die_visual_plot.py 使用plot可視化骰子
import matplotlib.pyplot as plt
from die import Die
# Initialization
die = Die()
# 擲骰子
results = [die.roll() for x in range(1000)]
# 分析結(jié)果
frequencies = [results.count(x) for x in range(1, die.num_sides+1)]
# 可視化結(jié)果
values = [x for x in range(1, die.num_sides+1)]
plt.plot(values, frequencies, linewidth=2, marker='o', markerfacecolor='yellow', markersize=5, color='b')
# 設(shè)置圖表
plt.title('Roll a die using matplotlib', fontsize=24)
plt.xlabel('Value', fontsize=14)
plt.ylabel('Frequency', fontsize=14)
# 顯示數(shù)據(jù)
for x, y in zip(values, frequencies):
  # 將y數(shù)據(jù)加載到(x,y)位置
  plt.text(x, y, y, fontsize=12, color='red', ha='center', va='bottom')
# 顯示結(jié)果
plt.show()

# die_visual_scatter.py 使用scatter可視化骰子
import matplotlib.pyplot as plt
from die import Die
# Initialization
die = Die()
# 擲骰子
results = [die.roll() for x in range(1000)]
# 分析結(jié)果
frequencies = [results.count(x) for x in range(1, die.num_sides+1)]
# 可視化結(jié)果
values = [x for x in range(1, die.num_sides+1)]
plt.scatter(values, frequencies, c=frequencies, cmap=plt.cm.Blues, edgecolor='none', s=10)
# 設(shè)置圖表
plt.title('Roll a die using matplotlib', fontsize=24)
plt.xlabel('Value', fontsize=14)
plt.ylabel('Frequency', fontsize=14)
# 顯示數(shù)據(jù)
for x, y in zip(values, frequencies):
  # 將y數(shù)據(jù)加載到(x,y)位置
  plt.text(x, y, y, fontsize=12, color='red', ha='center', va='bottom')
# 顯示結(jié)果
plt.show()