将数据导入散点图

问题描述

我有一个代码可以生成两种类型的卡片,每种可能的结果都显示 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()