How To Take Array As A Input In Python