Python Data Processing Using Arrays