使用python在极坐标中生成一条直线

"create a wall"
import numpy as np
import matplotlib.pyplot as plt
import sys

def main(distance):
    theta = np.arange(-45 / 180 * np.pi, 45 / 180 * np.pi, 1 / 180 * np.pi)

    tmp = np.cos(theta)


    wall = distance / tmp
    with open("dis.csv",mode = ‘w‘) as file:
        counter = 0
        s = ‘‘
        for n in wall:
            s += str("%d,"%n)
            counter += 1

        while counter < 360:
            s += "0,"
            counter += 1

        file.write(s)
        file.close()
    lin = np.linspace(distance+1000,distance+1000,len(theta))

    plt.polar(theta, wall)
    plt.polar(theta, lin)
    plt.show()

if __name__ == "__main__":
    main(int(sys.argv[1]))

python buildwall.py 3000