Python 使用matplotlib模块模拟掷骰子

最新推荐文章于 2020-11-20 23:52:50 发布

weixin_30451709

最新推荐文章于 2020-11-20 23:52:50 发布

阅读量352

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/noonjuan/p/10834410.html

博客主要围绕Python实现掷骰子展开，涉及骰子类的创建，还包括用折线图和散点图对掷骰子结果进行可视化展示，内容转载自相关博客。

掷骰子

骰子类

# die.py 骰子类模块

from random import randint

class Die():
    """骰子类"""

    def __init__(self, num_sides=6):
        """默认六面的骰子"""
        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)]

# 分析结果
frequencies = [results.count(x) for x in range(1, die.num_sides+1)]

# 可视化结果
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')

# 设置图表
plt.title('Roll a die using matplotlib', fontsize=24)
plt.xlabel('Value', fontsize=14)
plt.ylabel('Frequency', fontsize=14)

# 显示数据
for x, y in zip(values, frequencies):
    # 将y数据加载到(x,y)位置
    plt.text(x, y, y, fontsize=12, color='red', ha='center', va='bottom')

# 显示结果
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)]

# 分析结果
frequencies = [results.count(x) for x in range(1, die.num_sides+1)]

# 可视化结果
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)

# 设置图表
plt.title('Roll a die using matplotlib', fontsize=24)
plt.xlabel('Value', fontsize=14)
plt.ylabel('Frequency', fontsize=14)

# 显示数据
for x, y in zip(values, frequencies):
    # 将y数据加载到(x,y)位置
    plt.text(x, y, y, fontsize=12, color='red', ha='center', va='bottom')

# 显示结果
plt.show()