编写 python 代码时出现问题。在机器学习领域 [已结束]

问：

我有一个包含以下列的数据集：.如何修改我的代码，以便手动选择列而不是随机选择？您能否提供一个至少包含一个选定列的代码示例，以便我了解它应该是什么样子？ 我希望根据所选列构建图表，这些列将在代码中指定。

col_names = np.random.choice(train_x.columns, size=np.random.randint(1, len(train_x.columns)), replace=False)
print('Выбраны столбцы:\n', '\n'.join(col_names))
plt.figure(figsize=(20, 6))

for i, k in enumerate([3, 5, 10], 1):
    pred_y = knn(train_x, train_y, test_x, k, col_names)
    plot = plt.subplot(1, 3, i)
    plot.set_title('K = %d' % k)
    show_matrix(plot, pred_y, test_y, 3)