draw a polygon using vertices

plot_polygon.py

import matplotlib.pyplot as plt

def plot_polygon(vertices):
    plt.figure()
    for i in range(len(vertices)-1):
        vertex_set = [ vertices[i] ] + [ vertices[i+1] ]
        x, y = zip(*vertex_set)
        plt.plot(x,y)

    vertex_set = [ vertices[0] ] + [ vertices[-1] ]
    x, y = zip(*vertex_set)
    plt.plot(x,y)

    plt.xlim([min(vertices, key=lambda v: v[0])[0]-2, max(vertices, key=lambda v: v[0])[0]+2])
    plt.ylim([min(vertices, key=lambda v: v[1])[1]-2, max(vertices, key=lambda v: v[1])[1]+2])
    plt.show()
    

if __name__ == '__main__':
    vertices = [ (0,0), (0,1), (1,1), (1,0) ]
    plot_polygon(vertices)