Python. Pandas.Drop колонка

Question

У меня есть объект dataFrame.

df = pd.read_csv("new_data.csv", index_col = 0)

Когда я делаю

print df.head()

Вывод был

  ... Risk ...
0 ...  2   ...
1 ...  3   ...
...

Но когда я попробую это X = df.drop("Risk", 1).values

Произошла ошибка

"['Risk'] не найден на оси"

Answer 1

Если вы знаете позицию вашего столбца (среди столбцов), вы можете попытаться удалить свой столбец по индексу .

Предполагая, что вы хотитеудалить столбец № 3 (количество начинается с 0, но не включает индексный столбец), вы можете написать:

df.drop(df.columns[3], axis=1)

Приведенный выше код устойчив к любым «странным» или дополнительным символам в именах столбцов.

Или, может быть, вам стоит начать с print(df.columns)?Это покажет вам, какие имена столбцов в вашем DataFrame.