Python Generate Array Of Numbers