Python中等距取出一个数组其中n个数 函数实现

应用场景：

实验中不断得到新数据，想将数据图形化，但随着时间推移，数据越来越多，

此时需要我们等距选择数据列表中固定数量的数据，来进行图形化。

---

注：保留首尾数据。

import numpy as np

a = np.arange(0, 10)
b = np.arange(0, 15)
c = np.arange(0, 20)
d = np.arange(0, 30)
e = np.arange(10, 40)
f = np.arange(0, 100)


def pick_arange(arange, num):
    if num > len(arange):
        print("# num out of length, return arange:", end=" ")
        return arange
    else:
        output = np.array([], dtype=arange.dtype)
        seg = len(arange) / num
        for n in range(num):
            if int(seg * (n+1)) >= len(arange):
                output = np.append(output, arange[-1])
            else:
                output = np.append(output, arange[int(seg * n)])
        print("# return new arange:", end=' ')
        return output


print(pick_arange(a, 10))
print(pick_arange(a, 11))
print(pick_arange(b, 10))
print(pick_arange(c, 10))
print(pick_arange(d, 10))
print(pick_arange(e, 10))
print(pick_arange(f, 10))
print(pick_arange(f, 20))