How To Take Array Input In Python From User