Create An Array Using User Input In Python