我已经编写了一小段代码来模拟抛硬币的结果,并且想找到一种比连续抛硬币的列表更好的方式来呈现结果。作为我物理学学位的一部分,我已经学习 Python 一个月了,如果这有助于提供一些背景的话。
这是代码;
from pylab import *
x=0
while x<=100:
num = randint(0,2)
if num == 0:
print 'Heads'
else:
print 'Tails'
x=x+1
print 'Done'
我有哪些选择可以以更易于理解的方式呈现这些数据?
Python 无需使用 while
循环并将结果打印到屏幕,而是可以使用 Counter
非常巧妙地进行计数和存储结果。 ,内置字典容器的子类。
例如:
from collections import Counter
import random
Counter(random.choice(['H', 'T']) for _ in range(100))
当我运行代码时,它产生了以下计数:
Counter({'H': 52, 'T': 48})
我们可以看到正面被翻转了 52 次,反面被翻转了 48 次。
这已经很容易解释了,但是现在您已经有了数据结构中的数据,您还可以绘制一个简单的条形图。
遵循 Stack Overflow 答案中的建议 here ,你可以这样写:
import matplotlib.pyplot as plt
# tally = Counter({'H': 52, 'T': 48})
plt.bar(range(len(tally)), tally.values(), width=0.5, align='center')
plt.xticks(range(len(tally)), ['H', 'T'])
plt.show()
这会生成一个条形图,如下所示: