Code For 1d Array Python Questions