Доступ к столбцу в массиве numpy с помощью цикла for

Question

Я пытаюсь сохранить все столбцы массива в переменной с помощью цикла for (python 2). Однако существует либо синтаксическая ошибка, либо я должен определить x.

Это то, что я пробовал.

1)

x for x in train[:,x]:
  if x not in [target, IDcol]:
    predictors= x

2)

predictors = [x for x in train[:,x] if x not in [target, IDcol]]

Answer 1

Этот код не может работать.

Поскольку вы используете x как новую переменную, но вы уже используете ее в train[:,x]!

Измените ваши переменные:

predictors = [x2 for x2 in train[:,x] if x2 not in [target, IDcol]]