Как мне динамически добавлять столбцы во фрейм данных Python, чтобы я мог включать старшие члены для каждой переменной в моем фрейме данных?

Question

Я пытаюсь провести некоторую разработку функций для прогнозной модели. Сейчас у меня 55 переменных, и я хотел бы посмотреть на термины более высокого порядка для каждой переменной. Я хотел бы отрегулировать каждую переменную от квадрата до 15-й степени, а затем включить их в мой фрейм данных с именем столбца (т.е. var1_15). Я начал работать над этим в коде, но не увенчался успехом. Это правильный метод или есть лучший способ?

for k, i in range(2,16): 
    colname = 'k_%d'%i     
    dataframe[colname] = dataframe['k']**i
print(dataframe.head())

Answer 1

Я думаю, что это поможет:

i = 2
for col in df.columns:
    df['K_{}'.format(i)] = df[col] ** i
    i += 1

Кроме того, вы не можете использовать цикл for, если ваша интуиция в том, что нет разницы между i и k, нет причины использовать оба vars (вот почему также некоторая логика почему python не разрешает этого), если вам нужны 2 переменные и 2 итератора, рассмотрите возможность использования zip.

надеюсь, я помог