Coverting python list to smaller list or numpy array