Python 数据集切分：将数据划分为训练集和测试集

代码 train_data = data[:, :-3] 和 test_data = data[:, -3:] 用于将一个数据集 data 切分成训练集 train_data 和测试集 test_data。

代码解释如下：

• data 表示原始数据集，通常以 NumPy 数组的形式存储。
• [:, :-3] 表示取 data 中所有行的前 n-3 列，其中 n 表示 data 的列数。这些列将被用作训练数据。
• [:, -3:] 表示取 data 中所有行的最后 3 列。这些列将被用作测试数据。

示例

假设 data 具有 5 列，那么代码将执行以下操作：

• train_data 将包含 data 中的第 1 列到第 2 列（共 n-3=2 列）。
• test_data 将包含 data 中的第 3 列到第 5 列（共 3 列）。

总结

这段代码使用简单的切片操作将数据集切分成训练集和测试集，这是一种常用的数据预处理方法。这可以帮助我们建立机器学习模型并评估其性能。