Python Input Into Array Definition