使用numpy截取连续的测试集及剩余作为训练集

这个repo 用来记录一些python技巧、书籍、学习链接等，欢迎star

github地址

本来想从numpy中找一个可以将array格式的数据提取出连续的测试集，其余作为训练集的函数，但是没找到，就自己写了几个，记录之~

import numpy as np

"""
utils:
    用于截取连续的测试集及剩余作为训练集
"""

def trunc(arr, start, end):
    sample = arr[start:end]
    remainder = np.delete(arr, np.s_[start:end], axis=0)
    return sample, remainder

def trunc2