Change the KEY_A to KEY_ + number. For example: if Input.is_key_pressed(KEY_1)
For more inputs control, for example, print hello when number 1 OR 2 is pressed. Go to Project settings - Input Map, and you can create your own actions and what keys will trigger them. In code to check these actions: Input.action_pressed(YOURACTION)