How To Take Array As An Input In Python