Yes, you only want to assign income to New York that you earned while working there. New York will first tax all of your income earned, then reduce the tax based on the percentage of income earned in other states.
**Say "Thanks" by clicking the thumb icon in a post
**Mark the post that answers your question by clicking on "Mark as Best Answer"