Python 之 plotly 模拟掷骰子 学习笔记

from random import randint

class Die:
    # 表示一个骰子的类
    def __init__(self,num_sides=6):
        # 骰子默认6面
        self.num_sides = num_sides

    def roll(self):
        # 返回一个位于1和骰子之间的默认值
        return randint(1,self.num_sides)

from plotly.graph_objects import Bar,Layout
from plotly.offline import offline
from die import Die

# 创建两个 D6，同时掷两个同面骰子
die_1 = Die()
# die_2 = Die()
# 同时掷两个不同面骰子
die_2 = Die(10)

# 投100次骰子并存储在一个列表
results = []
for roll_num in range(5000):
    result = die_1.roll() + die_2.roll()
    results.append(result)
print(results)

# 分析结果
frequencies = []
max_result = die_1.num_sides + die_2.num_sides
for value in range(1,max_result):
    frequency = results.count(value)
    frequencies.append(frequency)

# 对结果进行可视化
x_values = list(range(1,max_result+1))
data = [Bar(x=x_values,y=frequencies)]

x_axis_config = {'title' : '结果','dtick' : 1}
y_axis_congig = {'title' : '结果的频率'}
my_layout = Layout(title='掷骰子1000次的结果',
                   xaxis=x_axis_config,yaxis=y_axis_congig)
offline.plot({'data' : data, 'layout' : my_layout }, filename = 'd6_d6.html')


print(frequencies)