함수를 사용하여 팬더 DataFrame 열 이름 수정하기

Question

열 이름을 정리하기 위해 각 DataFrames에 적용 할 수있는 함수를 만들고 싶습니다. 예를 들어

:

def standardize_headers(df): 
    return df.columns.str.replace(' ', '_').str.lower() 

는 그러나, 나는이 코드를 사용하여 적용 할 때 :

이

AttributeError: ("'Series' object has no attribute 'columns'", u'occurred at index External_Code') 

'External_Code'는 첫 번째 열입니다 :

df2 = df.apply(standardize_headers) 

을 나는 오류 df에 이 간단한 예제를 람다 함수로 처리 할 수 ​​있다는 것을 알고 있지만 여러 단계로 더 복잡한 함수를 사용할 계획이므로 사용자 정의 함수에서 열 이름을 참조 할 수 있어야합니다. 나는 적용이 그 대상을 시리즈로 취급한다는 것을 안다. 그렇다면 사용자 정의 함수 내에서 열을 어떻게 참조 할 수 있습니까? 또는 문제는 내가 DataFrame 함수를 적용하는 방식입니다.

Answer 1

대신을 시도해보십시오

def standardize_headers(df, func=None): 
    df.columns = df.columns.str.replace(' ', '_').str.lower() 
    if func: 
     df = df.apply(func) 
    return df 

df2 = standardize_headers(df) 

당신도 단순히 이름 DataFrame을 반환하려면이 템플릿을 확장 할 수 있습니다, 또는 그것을 반환하기 전에 DataFrame에 적용 할 다양한 기능을 전달합니다.

Answer 2

간단한 방법은 rename 방법을 사용하는 것입니다. columns 매개 변수에 함수를 전달할 수 있으며이 함수는 각 열 이름에 적용됩니다. 이 함수의 x은 문자열이므로 직접 문자열 메서드를 사용할 수 있습니다.

def standardize_headers(x): 
    return x.replace(' ', '_').lower() 

df.rename(columns=standardize_headers) 

apply 함수는 각 열의 값을 Series로 직접 사용합니다.