Python Making Array Of Certain Length