Python浪漫之随机绘制不同颜色的气球

目录

1、效果图展示

2、完整代码

---

1、效果图展示

提示： 颜色和位置都是随机的，但是气球大小都是一致的

评论中记得分享你认为最好看的气球图出来吧！（欢迎参与）

2、完整代码

import matplotlib.pyplot as plt
import numpy as np


def draw_balloon(ax, x, y, color='red'):
    # 绘制气球的圆形部分
    balloon = plt.Circle((x, y), 0.5, color=color, ec='black')
    ax.add_patch(balloon)

    # 绘制气球的绳子
    ax.plot([x, x], [y - 0.5, y - 2], color='black', linewidth=2)


def draw_balloons(num_balloons=3):
    fig, ax = plt.subplots(figsize=(8, 8))
    ax.set_xlim(-2, 2)
    ax.set_ylim(-3, 1)
    ax.set_aspect('equal')

    # 随机生成气球的位置和颜色
    colors = ['red', 'blue', 'green', 'yellow', 'purple', 'orange']
    for _ in range(num_balloons):
        x = np.random.uniform(-1.5, 1.5)  # 随机x坐标
        y = np.random.uniform(0, 0.5)  # 随机y坐标
        color = np.random.choice(colors)  # 随机选择颜色
        draw_balloon(ax, x, y, color)

    ax.set_title('气球', fontsize=16)
    ax.axis('off')  # 关闭坐标轴
    plt.show()


def main():
    draw_balloons(num_balloons=5)  # 绘制5个气球


if __name__ == "__main__":
    main()