截取数据集dataset的最后两列,赋值给数据集train_data,提供程序示例
时间: 2024-03-04 22:48:07 浏览: 220
train set-数据集
假设数据集dataset是一个二维数组,可以使用以下代码来截取最后两列并赋值给train_data:
```python
import numpy as np
# 假设dataset是一个二维数组,最后两列是需要截取的
last_two_cols = dataset[:, -2:]
# 将截取的数据赋值给train_data
train_data = np.array(last_two_cols)
```
这段代码使用numpy库的切片语法,将最后两列截取出来,然后使用numpy的array()方法将其转换为一个新的二维数组train_data。
阅读全文