需要在图中使用 matplotlib 计算 LHS 和 RHS 点的框

Question

import random
import matplotlib.pyplot as plt

num1 = 30
num3 = [4000,3000,1500,9000,2500,8000,1200,800,900,1000,5400,9500,1100,3400,8100,
        5500,1200,3830,2311,9999]
num2 = []
for i in range (len(num3)):
    num2.append(random.randrange(0,45))
print(num2)

plt.axvline(num1,0,color="r")
plt.scatter(num2, num3)

代码运行良好。没有问题。我想知道如何从下图中得到与它所说的完全相同的盒子：

运行代码后，您将看到一个类似的图像，但没有该框。红线上的点被认为是在 RHS 中。我使用 excel 制作了这张图来显示我想要的内容。而且，你只能使用 matplotlib。

我可能需要下面的代码来计数

LHS_count = 0
for value in num2:
    if value < num1:
        LHS_count += 1
        print(LHS_count)
RHS_count = 0
for value in num2:
    if value >= num1:
        RHS_count += 1
        print(RHS_count)

Answer 1

我会推荐使用 numpy 来做类似的事情

import matplotlib.pyplot as plt
import numpy as np

vl_pos = 30
y = np.array([4000,3000,1500,9000,2500,8000,1200,800,900,1000,5400,9500,1100,3400,8100,
        5500,1200,3830,2311,9999])

x = np.random.random(y.shape) * 45

lhs = (x < vl_pos).sum()
rhs = (x >= vl_pos).sum()
lhs,rhs

txt = ("lhs: %s\n"
      "rhs: %s"%(lhs,rhs))

f,a = plt.subplots()
a.axvline(vl_pos,color="r")
a.scatter(x, y)
bbox =dict(facecolor = 'w')
a.text(0.8, 0.8, txt, bbox = bbox, transform = a.transAxes)