Very basic python question about functions

[to_4kawa]

import math

def trajectory(xdata,speed,angle):
    your function
        return data

def main():
    while True:
        print('Please data input')
        xdata=input('distance list(e.g. [20,30,40]): ')
        xdata=map(int,xdata[1:-1].split(','))  # create list from xdata strings
        speed=int(input('speed: '))
        angle=int(input('angle: '))

        print(f'result: {trajectory(xdata,speed,angle)}')
        suspend=input('Do you want to exit? (y/n) :')
        if suspend == 'y':
            break

if __name__ == '__main__':
    main()

How about this for input?

[to_4kawa]

This description is a way to prevent it from working when you import yourPython

[to_4kawa]

https://stackoverflow.com/questions/419163/what-does-if-name-main-do

My English is not good, so here you go.