编写一个函数,将list类型数据集的最后五十个样本分为测试样本,其余的作为训练样本
时间: 2023-06-08 17:04:52 浏览: 51
您好!对于您的问题,我可以提供以下Python代码来实现最后50个样本作为测试集的操作:
```python
def train_test_split(data):
"""
将list类型数据集的最后50个样本分为测试样本,其余的作为训练样本
"""
train_data = data[:-50]
test_data = data[-50:]
return train_data, test_data
```
其中,参数`data`是list类型的数据集,函数将其最后50个样本分为测试样本,其余的作为训练样本,并分别以两个list的形式返回。例如,可以这样使用该函数:
```python
data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20,
21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40]
train_data, test_data = train_test_split(data)
print(train_data)
print(test_data)
```
输出结果如下:
```
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30]
[31, 32, 33, 34, 35, 36, 37, 38, 39, 40]
```
希望能够帮到您!如果您还有其他问题,欢迎继续提问!