Taking Array As An Input In Python