Python + Pandas + Dataframe: создание динамического имени столбца для столбца Pandas dataframe - PullRequest
0 голосов
/ 05 апреля 2019
data['Predicted Order'] = predicted_order
print(data)
data.to_csv('smallys_ARMA.csv', encoding='utf-8', index=False)

Кадр данных Pandas data получает новый столбец Predicted Order в кадре данных, значения которого назначены списку predicted_order.

Здесь я не хочу жестко кодировать недавно сгенерированное имя столбца, теперь оно Predicted Order. Я хочу, чтобы каждый новый сгенерированный столбец получал имя динамически, как это Week_i для i = 0,1,2 и т. Д.

Edit: Этот код будет выполняться неизвестное количество раз в будущем, и каждый раз он будет добавлять новый столбец в фрейм данных.

Ответы [ 2 ]

0 голосов
/ 05 апреля 2019

может быть такая вещь:

numbers = range(52)
for i in numbers:
    if not "weeks_{i}" in df.columns:
        df["weeks_{i}"] = predicted_order
        break
0 голосов
/ 05 апреля 2019

Для этого вы можете использовать форматирование строки.

weeks = range(42)

for i in weeks:
     data[f'Week_{i}'] = predicted_order

Если вы используете Python <3.6, используйте <code>data['Week_{}'.format(i)] вместо

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...