selected_feature = [0, 1, 2]
data_set = np.array([[1, 2, 3, 4, 5], [6, 7, 8 ,9, 10]])
print(data_set[:, selected_feature])
打印结果:
[[1 2 3]
[6 7 8]]
selected_feature = [0, 1, 2]
data_set = np.array([[1, 2, 3, 4, 5], [6, 7, 8 ,9, 10]])
print(data_set[:, selected_feature])
打印结果:
[[1 2 3]
[6 7 8]]