问题描述
我有一个代码可以生成两种类型的卡片,每种可能的结果都显示 6-10 张卡片。代码如下:
from sequence import deckOrder
for i in range(len(deckOrder)):
for j in range(len(deckOrder)):
#Draw
first_draw = deckOrder[i]
second_draw = deckOrder[j]
#Before first draw
initial_estimate = 0.5
initial_variance = 1/12
#After first draw
r_count = first_draw.count('R')
b_count = first_draw.count('B')
alpha = 1 + r_count
beta = 1 + b_count
e_of_theta = alpha/(alpha+beta)
surprise = ((e_of_theta - initial_estimate)**2)/(initial_variance)
var_theta = (alpha * beta)/ ((alpha + beta) **2 *(alpha + beta + 1))
#After second draw
r_count = second_draw.count('R')
b_count = second_draw.count('B')
new_alpha = alpha + r_count
new_beta = beta + b_count
new_e_of_theta = new_alpha/(new_alpha + new_beta)
surprise = ((new_e_of_theta - e_of_theta)**2)/var_theta
#Result
new_list = [surprise]
我把数据放到了列表中,但是不知道怎么把这段代码生成的结果放到散点图中。有人可以帮我解决这个问题吗?
解决方法
要将简单数字放入散点图中,请定义 x 和 y 坐标:
import matplotlib.pyplot as plt
import numpy as np
x = np.array([1,2,3])
y = np.array([50,51,52])
plt.scatter(x,y)
plt.show()