Как я могу использовать цикл for для создания квадратов версий существующих столбцов?

Question

Я использую набор данных Hitters.csv и пытаюсь создать квадратные версии существующих количественных столбцов, используя цикл for.Я создал набор данных всех количественных столбцов в Hitters.csv:

features_df = hitters_df.drop(['Salary', 'CAtBat', 'CHits', 'CHmRun', 'CRBI', 'CRuns', 'CWalks'], axis=1)

Я новичок в Python и неопытный с циклами.Вот что у меня есть:

for i in features_df:
    print (i**2)

Как мне сделать эту работу?Нужно ли использовать словарь?

Answer 1

Wrt «создать квадратные версии существующих количественных столбцов» , если вы отбросили все нечисловые столбцы, вы можете сделать:

features_df = features_df.apply(np.square)
# or just
features_df = features_df ** 2

Если вы не удалили все нечисловые столбцы , а ваш hitters_df также включает столбцы с текстовыми данными (т. Е. Dtype * object), то переберите столбцы, пропуская ошибки или проверяя, является ли этодопустимый тип.Или, что еще лучше, переберите только числовые столбцы :

for column in features_df.select_dtypes(include='number').columns:
    features_df[column] = features_df[column] ** 2