• 【科技与狠活】如何利用Python绘制足球场

     

    卡塔尔世界杯赛程近半，朋友圈都在晒中奖的体育彩票，而我在搬砖🧱。

    今天我将介绍如何使用Python Matplotlib创建一个足球场，本文设计球场尺寸为105×68。

    首先导入所需的依赖包：

    1. import pandas as pd
    2. import numpy as np
    3.  
    4. import matplotlib.pyplot as plt
    5. from matplotlib.patches import Arc

    其实看似复杂的足球场一旦将其分解成各个组成部分，就不是什么特别复杂的事情了。足球场只是线条和形状的集合，下面我将一步步实现其包含的各个元素并在最后将它们全部封装在一个函数中。

    首先，设置一些会频繁用到的全局变量。

    1. # 足球场的边界长宽
    2. x_min = 0
    3. x_max = 105
    4. y_min = 0
    5. y_max = 68
    6.  
    7. line_color = "grey"
    8. line_thickness = 1.5
    9. background = "w" # w表示white
    10. point_size = 20
    11.  
    12. arc_angle = 0
    13. first = 0
    14. second = 1

    接下来创建列表pitch_x、pitch_y标记球场上各个元素的坐标。

    1. pitch_x = [0,5.8,11.5,17,50,83,88.5,94.2,100] # pitch x markings
    2.  
    3. """
    4. [goal line, six yard box, penalty spot, edge of box,
    5.  halfway line, edge of box, penalty spot, six yard box, goal line]
    6. """
    7.  
    8. pitch_y = [0, 21.1, 36.6, 50, 63.2, 78.9, 100]
    9.  
    10. """
    11. [sideline, edge of box, six yard box, centre of pitch,
    12.  six yard box, edge of box, sideline ]
    13. """
    14.  
    15. goal_y = [45.2, 54.8] # goal posts

    上述定义的x、y为100x100, 需要对球场进行同比例转换为105x68，当然球场元素坐标也会同比例变化：

    1. x_conversion = 105 / 100
    2. y_conversion = 68 / 100
    3.  
    4. pitch_x = [item * x_conversion for item in pitch_x]
    5. pitch_y = [item * y_conversion for item in pitch_y]
    6. goal_y = [item * y_conversion for item in goal_y]

    现在已经对元素的点坐标进行了转换，下面就可以通过 x 和 y 坐标列表来绘制足球场的各个元素了。例如，这是如何绘制球场边线的代码：

    1. lx1 = [x_min, x_max, x_max, x_min, x_min]
    2. ly1 = [y_min, y_min, y_max, y_max, y_min]
    3.  
    4. fig, ax = plt.subplots(figsize=(11,7))
    5. ax.axis("off")
    6. ax.plot(lx1,ly1, color=line_color, lw=line_thickness, zorder=-1)
    7.  
    8. plt.tight_layout()
    9. plt.show()

    c7a98a47714d3c362d9e3adf4517f7ff.png

    下面可以遍历每个元素的标记并将它们绘制成对应的球场元素。

    1. # side and goal lines
    2. lx1 = [x_min, x_max, x_max, x_min, x_min]
    3. ly1 = [y_min, y_min, y_max, y_max, y_min]
    4.  
    5. ax.plot(lx1, ly1, color=line_color, lw=line_thickness, zorder=-1)
    6.  
    7. # outer boxed
    8. lx2 = [x_max, pitch_x[5], pitch_x[5], x_max]
    9. ly2 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    10.  
    11. ax.plot(lx2, ly2, color=line_color, lw=line_thickness, zorder=-1)
    12.  
    13. lx3 = [0, pitch_x[3], pitch_x[3], 0]
    14. ly3 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    15.  
    16. ax.plot(lx3, ly3, color=line_color, lw=line_thickness, zorder=-1)
    17.  
    18. # .....

    当然这不是最好的方法，因为这意味着如果我想绘制垂直方向足球场，我需要调整每一项数据。与其一一绘制，不如构建坐标列表并将它们添加到主列表中：

    1. # side and goal lines
    2. lx1 = [x_min, x_max, x_max, x_min, x_min]
    3. ly1 = [y_min, y_min, y_max, y_max, y_min]
    4.  
    5. # outer boxed
    6. lx2 = [x_max, pitch_x[5], pitch_x[5], x_max]
    7. ly2 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    8.  
    9. lx3 = [0, pitch_x[3], pitch_x[3], 0]
    10. ly3 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    11.  
    12. # goals
    13. lx4 = [x_max, x_max+2, x_max+2, x_max]
    14. ly4 = [goal_y[0], goal_y[0], goal_y[1], goal_y[1]]
    15.  
    16. lx5 = [0, -2, -2, 0]
    17. ly5 = [goal_y[0], goal_y[0], goal_y[1], goal_y[1]]
    18.  
    19. # 6 yard boxes
    20. lx6 = [x_max, pitch_x[7], pitch_x[7], x_max]
    21. ly6 = [pitch_y[2], pitch_y[2], pitch_y[4], pitch_y[4]]
    22.  
    23. lx7 = [0, pitch_x[1], pitch_x[1], 0]
    24. ly7 = [pitch_y[2], pitch_y[2], pitch_y[4], pitch_y[4]]
    25.  
    26. # Halfway line, penalty spots, and kickoff spot
    27. lx8 = [pitch_x[4], pitch_x[4]]
    28. ly8 = [0, y_max]
    29.  
    30. lines = [
    31.     [lx1, ly1],
    32.     [lx2, ly2],
    33.     [lx3, ly3],
    34.     [lx4, ly4],
    35.     [lx5, ly5],
    36.     [lx6, ly6],
    37.     [lx7, ly7],
    38.     [lx8, ly8],
    39.     ]

    通过循环遍历lines列表来绘制上面的图：

    1. fig, ax = plt.subplots(figsize=(11,7))
    2. ax.axis("off") 
    3.  
    4. for line in lines:
    5.     ax.plot(line[0], line[1],
    6.             color=line_color,
    7.             lw=line_thickness,
    8.             zorder=-1)
    9.  
    10. plt.tight_layout()
    11. plt.show()

    2dc85617762e170dc8fdc2653b72d33c.png

    这样绘图代码看起来更清晰，以同样的方式绘制开球点和罚球点：

    1. points = [
    2.     [pitch_x[6], pitch_y[3]],
    3.     [pitch_x[2], pitch_y[3]],
    4.     [pitch_x[4], pitch_y[3]]
    5.     ]

    然后遍历绘图代码中的点:

    1. fig, ax = plt.subplots(figsize=(11,7))
    2. ax.axis("off")
    3.  
    4. for line in lines:
    5.     ax.plot(line[0], line[1],
    6.             color=line_color,
    7.             lw=line_thickness,
    8.             zorder=-1)
    9.  
    10. for point in points:
    11.     ax.scatter(point[0], point[1],
    12.                color=line_color,
    13.                s=point_size,
    14.                zorder=-1)
    15.  
    16. plt.tight_layout()
    17. plt.show()

    0b127daacc9707da62d6dcec68715e24.png

    在每个"box"上添加中心圆。

    1. circle_points = [pitch_x[4], pitch_y[3]]
    2. arc_points1 = [pitch_x[6], pitch_y[3]]
    3. arc_points2 = [pitch_x[2], pitch_y[3]]
    4. fig, ax = plt.subplots(figsize=(11,7))
    5. ax.axis("off")
    6. for line in lines:
    7.     ax.plot(line[0], line[1],
    8.             color=line_color,
    9.             lw=line_thickness,
    10.             zorder=-1)
    11.  
    12. for point in points:
    13.     ax.scatter(point[0], point[1],
    14.                color=line_color,
    15.                s=point_size,
    16.                zorder=-1)
    17.  
    18. circle = plt.Circle((circle_points[0], circle_points[1]),
    19.                      x_max * 0.088,
    20.                      lw=line_thickness,
    21.                      color=line_color,
    22.                      fill=False,
    23.                      zorder=-1)
    24.  
    25. ax.add_artist(circle)
    26.  
    27. arc1 = Arc((arc_points1[0], arc_points1[1]),
    28.            height=x_max * 0.088 * 2,
    29.            width=x_max * 0.088 * 2,
    30.            angle=arc_angle,
    31.            theta1=128.75,
    32.            theta2=231.25,
    33.            color=line_color,
    34.            lw=line_thickness,
    35.            zorder=-1)
    36.  
    37. ax.add_artist(arc1)
    38.  
    39. arc2 = Arc((arc_points2[0], arc_points2[1]),
    40.            height=x_max * 0.088 * 2,
    41.            width=x_max * 0.088 * 2,
    42.            angle=arc_angle,
    43.            theta1=308.75,
    44.            theta2=51.25,
    45.            color=line_color,
    46.            lw=line_thickness,
    47.            zorder=-1)
    48.  
    49. ax.add_artist(arc2)
    50.  
    51. ax.set_aspect("equal") 
    52. plt.tight_layout()
    53. plt.show()

    737562edc08e1311c7b23901deed51bb.png

    这就是创建足球场的整个过程！然而，这还没有结束。我添加了几个变量并调整了代码以绘制逻辑更加灵活。

    1. fig, ax = plt.subplots(figsize=(11,7))
    2. ax.axis("off")
    3.  
    4. for line in lines:
    5.     ax.plot(line[first], line[second],
    6.             color=line_color,
    7.             lw=line_thickness,
    8.             zorder=-1)
    9.  
    10. for point in points:
    11.     ax.scatter(point[first], point[second],
    12.                color=line_color,
    13.                s=point_size,
    14.                zorder=-1)
    15.  
    16. circle = plt.Circle((circle_points[first], circle_points[second]),
    17.                      x_max * 0.088,
    18.                      lw=line_thickness,
    19.                      color=line_color,
    20.                      fill=False,
    21.                      zorder=-1)
    22.  
    23. ax.add_artist(circle)
    24.  
    25. arc1 = Arc((arc_points1[first], arc_points1[second]),
    26.            height=x_max * 0.088 * 2,
    27.            width=x_max * 0.088 * 2,
    28.            angle=arc_angle,
    29.            theta1=128.75,
    30.            theta2=231.25,
    31.            color=line_color,
    32.            lw=line_thickness,
    33.            zorder=-1)
    34.  
    35. ax.add_artist(arc1)
    36.  
    37. arc2 = Arc((arc_points2[first], arc_points2[second]),
    38.            height=x_max * 0.088 * 2,
    39.            width=x_max * 0.088 * 2,
    40.            angle=arc_angle,
    41.            theta1=308.75,
    42.            theta2=51.25,
    43.            color=line_color,
    44.            lw=line_thickness,
    45.            zorder=-1)
    46.  
    47. ax.add_artist(arc2)
    48.  
    49. ax.set_aspect("equal")
    50. plt.tight_layout()
    51. plt.show()

    还记得文章开头设置的几个变量吧：arc_angle = 0、first = 0 和 second = 1。通过将这些设置为变量而不是硬编码，我们可以修改这几个变量为first = 1、second = 0和arc_angle = 90，这样就可以绘制一个垂直方向的足球场了。

    1. first = 1
    2. second = 0
    3. arc_angle = 90
    4.  
    5. fig, ax = plt.subplots(figsize=(7,11))
    6. ax.axis("off")
    7.  
    8. for line in lines:
    9.     ax.plot(line[first], line[second],
    10.             color=line_color,
    11.             lw=line_thickness,
    12.             zorder=-1)
    13.  
    14. for point in points:
    15.     ax.scatter(point[first], point[second],
    16.                color=line_color,
    17.                s=point_size,
    18.                zorder=-1)
    19.  
    20. circle = plt.Circle((circle_points[first], circle_points[second]),
    21.                      x_max * 0.088,
    22.                      lw=line_thickness,
    23.                      color=line_color,
    24.                      fill=False,
    25.                      zorder=-1)
    26.  
    27. ax.add_artist(circle)
    28.  
    29. arc1 = Arc((arc_points1[first], arc_points1[second]),
    30.            height=x_max * 0.088 * 2,
    31.            width=x_max * 0.088 * 2,
    32.            angle=arc_angle,
    33.            theta1=128.75,
    34.            theta2=231.25,
    35.            color=line_color,
    36.            lw=line_thickness,
    37.            zorder=-1)
    38.  
    39. ax.add_artist(arc1)
    40.  
    41. arc2 = Arc((arc_points2[first], arc_points2[second]),
    42.            height=x_max * 0.088 * 2,
    43.            width=x_max * 0.088 * 2,
    44.            angle=arc_angle,
    45.            theta1=308.75,
    46.            theta2=51.25,
    47.            color=line_color,
    48.            lw=line_thickness,
    49.            zorder=-1)
    50.  
    51. ax.add_artist(arc2)
    52.  
    53. ax.set_aspect("equal")
    54. plt.tight_layout()
    55. plt.show()

    b2d1e5f8596e052b8c1cad6e85532c3f.png

    代码封装

    下面将上面零散的代码封装成函数。

    1. def draw_pitch(x_min=0, x_max=105,
    2.                y_min=0, y_max=68,
    3.                pitch_color="w",
    4.                line_color="grey",
    5.                line_thickness=1.5,
    6.                point_size=20,
    7.                orientation="horizontal",
    8.                aspect="full",
    9.                ax=None
    10.                ):
    11.  
    12.     if not ax:
    13.         raise TypeError("This function is intended to be used with an existing fig and ax in order to allow flexibility in plotting of various sizes and in subplots.")
    14.  
    15.  
    16.     if orientation.lower().startswith("h"):
    17.         first = 0
    18.         second = 1
    19.         arc_angle = 0
    20.  
    21.         if aspect == "half":
    22.             ax.set_xlim(x_max / 2, x_max + 5)
    23.  
    24.     elif orientation.lower().startswith("v"):
    25.         first = 1
    26.         second = 0
    27.         arc_angle = 90
    28.  
    29.         if aspect == "half":
    30.             ax.set_ylim(x_max / 2, x_max + 5)
    31.  
    32.     
    33.     else:
    34.         raise NameError("You must choose one of horizontal or vertical")
    35.  
    36.     
    37.     ax.axis("off")
    38.  
    39.     rect = plt.Rectangle((x_min, y_min),
    40.                          x_max, y_max,
    41.                          facecolor=pitch_color,
    42.                          edgecolor="none",
    43.                          zorder=-2)
    44.  
    45.     ax.add_artist(rect)
    46.  
    47.     x_conversion = x_max / 100
    48.     y_conversion = y_max / 100
    49.  
    50.     pitch_x = [0,5.8,11.5,17,50,83,88.5,94.2,100] # pitch x markings
    51.     pitch_x = [x * x_conversion for x in pitch_x]
    52.  
    53.     pitch_y = [0, 21.1, 36.6, 50, 63.2, 78.9, 100] # pitch y markings
    54.     pitch_y = [x * y_conversion for x in pitch_y]
    55.  
    56.     goal_y = [45.2, 54.8] # goal posts
    57.     goal_y = [x * y_conversion for x in goal_y]
    58.  
    59.     # side and goal lines
    60.     lx1 = [x_min, x_max, x_max, x_min, x_min]
    61.     ly1 = [y_min, y_min, y_max, y_max, y_min]
    62.  
    63.     # outer boxed
    64.     lx2 = [x_max, pitch_x[5], pitch_x[5], x_max]
    65.     ly2 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    66.  
    67.     lx3 = [0, pitch_x[3], pitch_x[3], 0]
    68.     ly3 = [pitch_y[1], pitch_y[1], pitch_y[5], pitch_y[5]]
    69.  
    70.     # goals
    71.     lx4 = [x_max, x_max+2, x_max+2, x_max]
    72.     ly4 = [goal_y[0], goal_y[0], goal_y[1], goal_y[1]]
    73.  
    74.     lx5 = [0, -2, -2, 0]
    75.     ly5 = [goal_y[0], goal_y[0], goal_y[1], goal_y[1]]
    76.  
    77.     # 6 yard boxes
    78.     lx6 = [x_max, pitch_x[7], pitch_x[7], x_max]
    79.     ly6 = [pitch_y[2],pitch_y[2], pitch_y[4], pitch_y[4]]
    80.  
    81.     lx7 = [0, pitch_x[1], pitch_x[1], 0]
    82.     ly7 = [pitch_y[2],pitch_y[2], pitch_y[4], pitch_y[4]]
    83.  
    84.  
    85.     # Halfway line, penalty spots, and kickoff spot
    86.     lx8 = [pitch_x[4], pitch_x[4]]
    87.     ly8 = [0, y_max]
    88.  
    89.     lines = [
    90.         [lx1, ly1],
    91.         [lx2, ly2],
    92.         [lx3, ly3],
    93.         [lx4, ly4],
    94.         [lx5, ly5],
    95.         [lx6, ly6],
    96.         [lx7, ly7],
    97.         [lx8, ly8],
    98.         ]
    99.  
    100.     points = [
    101.         [pitch_x[6], pitch_y[3]],
    102.         [pitch_x[2], pitch_y[3]],
    103.         [pitch_x[4], pitch_y[3]]
    104.         ]
    105.  
    106.     circle_points = [pitch_x[4], pitch_y[3]]
    107.     arc_points1 = [pitch_x[6], pitch_y[3]]
    108.     arc_points2 = [pitch_x[2], pitch_y[3]]
    109.  
    110.  
    111.     for line in lines:
    112.         ax.plot(line[first], line[second],
    113.                 color=line_color,
    114.                 lw=line_thickness,
    115.                 zorder=-1)
    116.  
    117.     for point in points:
    118.         ax.scatter(point[first], point[second],
    119.                    color=line_color,
    120.                    s=point_size,
    121.                    zorder=-1)
    122.  
    123.     circle = plt.Circle((circle_points[first], circle_points[second]),
    124.                         x_max * 0.088,
    125.                         lw=line_thickness,
    126.                         color=line_color,
    127.                         fill=False,
    128.                         zorder=-1)
    129.  
    130.     ax.add_artist(circle)
    131.  
    132.     arc1 = Arc((arc_points1[first], arc_points1[second]),
    133.                height=x_max * 0.088 * 2,
    134.                width=x_max * 0.088 * 2,
    135.                angle=arc_angle,
    136.                theta1=128.75,
    137.                theta2=231.25,
    138.                color=line_color,
    139.                lw=line_thickness,
    140.                zorder=-1)
    141.  
    142.     ax.add_artist(arc1)
    143.  
    144.     arc2 = Arc((arc_points2[first], arc_points2[second]),
    145.                height=x_max * 0.088 * 2,
    146.                width=x_max * 0.088 * 2,
    147.                angle=arc_angle,
    148.                theta1=308.75,
    149.                theta2=51.25,
    150.                color=line_color,
    151.                lw=line_thickness,
    152.                zorder=-1)
    153.  
    154.     ax.add_artist(arc2)
    155.  
    156.     ax.set_aspect("equal")
    157.  
    158.     return ax

    现在只需要调用函数来绘制想要的足球场！

    1. background = "#09b309"
    2.  
    3. fig, ax = plt.subplots(figsize=(11, 7))
    4. fig.set_facecolor(background)
    5.  
    6. draw_pitch(orientation="h",
    7.            aspect="full",
    8.            pitch_color=background, 
    9.            line_color="lightgrey",
    10.            ax=ax)
    11.  
    12. plt.tight_layout()
    13. plt.show()

    db80e051ac7dcac3661c1c30082ff516.png

     

    下面这张图是绘制过程gif。

    8618f5759408690f5a5fb5e4dbd48099.gif



    代码仓库：How-To-Python: 人生苦短，我用Python

     

     

  • 相关阅读:
    npm运行vue项目出现禁止运行脚本
    3-2数据链路层-差错控制
    Day802.JVM热点问题 -Java 性能调优实战
    jira查询user详细信息
    【案例实战】SpringBoot整合Redis连接池生成图形验证码
    Visual Studio 2022开发Arduino详述
    云计算基础技术
    【Codeforces Round #811 (Div. 3)】【题目解析+AK代码】
    Rust开发——闭包使用示例
    N-HiTS: Neural Hierarchical Interpolation for Time Series Forecasting
  • 原文地址：https://blog.csdn.net/csd11311/article/details/128178369