Python Create Array With Steps