https://tutorial.eyehunts.com/python/how-to-take-array-input-in-python-example-code/
How to take array input in Python | Example code