добавить столбец данных моей функции в исходные данные в Python

Question

Я пытаюсь добавить значение своей функции в столбец набора данных.У меня есть восемь столбцов:

'DATE','Max_R','Total_R','Avg_R','MAX_T','TOTAL_T','AVG_T'

Затем я разделил свой столбец DATE на три столбца: День, Месяц и Год соответственно.Вот мой код на python:

import pandas as pd
import numpy as np

df=pd.read_csv('moving_average_calculation.csv', sep=',')
#df = pd.DataFrame(columns=['DATE','Max_R','Total_R','Avg_R','MAX_T','TOTAL_T','AVG_T'])
df = pd.DataFrame(pd.date_range('1-Jan-08', periods=2558),columns=['DATE'])


def f(df):
    df = df.copy()
    df['Day'] = pd.DatetimeIndex(df['DATE']).day
    df['Month'] =pd.DatetimeIndex(df['DATE']).month
    df['Year'] = pd.DatetimeIndex(df['DATE']).year
    return df

print(f(df).head(10))

Теперь я хочу получить мои столбцы, которые будут иметь эти столбцы:

'Day','Month','Year','Max_R','Total_R','Avg_R','MAX_T','TOTAL_T','AVG_T'

Как я могу это сделать?Спасибо.

Answer 1

Ваш вопрос немного неясен, так как вы определяете df дважды (+1 в комментарии), но если я правильно понимаю (то есть: у вас уже есть «DATE» в файле .csv), это может помочь:

df = pd.read_csv('moving_average_calculation.csv', sep=',')

df['Day'] = pd.DatetimeIndex(df['DATE']).day
df['Month'] = pd.DatetimeIndex(df['DATE']).month
df['Year'] = pd.DatetimeIndex(df['DATE']).year

df.drop('DATE', axis=1, inplace=True)

df = df[['Day','Month','Year','Max_R','Total_R','Avg_R','MAX_T','TOTAL_T','AVG_T']]