figutils.py 文件源码

def extract_n_elements(l, n):
    while not gcd(n, len(l)) == n:
        l.pop(0)
    len_div = len(l)
    step = len_div // n
    ids = [i - 1 for i in np.arange(step, len_div + step, step)]  # -1 for indexing
    elements = np.asarray(l)[ids].tolist()
    return elements