python利用蒙特卡洛方法绘制PI的示意图

利用python的numpy和matplotlib库

import matplotlib.pyplot as plt
from matplotlib.patches import Circle
import numpy as np

n = 5000  # 投掷10000个点
blue_x = []
blue_y = []
red_x = []
red_y = []
for i in range(1, n):
    x = np.random.uniform(0, 1, 1)
    y = np.random.uniform(0, 1, 1)
    distance = pow(x ** 2 + y ** 2, 0.5)
    if distance < 1:
        blue_x.append(x), blue_y.append(y)
    else:
        red_x.append(x), red_y.append(y)
fig = plt.subplots(1, 1)
plt.scatter(blue_x, blue_y, color='b')
plt.scatter(red_x, red_y, color='r')
plt.xlabel('X')
plt.ylabel('Y')
plt.xlim((0, 1))  # 修改坐标轴范围
plt.ylim((0, 1))
fig = plt.gca()
fig.set_aspect(1)
# 画圆
circle = Circle((0, 0), radius=1, facecolor='none', edgecolor='black', linewidth=3, alpha=1)
fig.add_patch(circle)
# 显示图像
plt.show()

猜你喜欢

转载自blog.csdn.net/s1hjf/article/details/127032741