在Python中等距取出一个数组其中n个数的实现方式
时间：2020-11-23 11:45:37 | 栏目：Python代码 | 点击：次
应用场景：
实验中不断得到新数据，想将数据图形化，但随着时间推移，数据越来越多，
此时需要我们等距选择数据列表中固定数量的数据，来进行图形化。
注：保留首尾数据。
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))
