関数を使用してPandas DataFrameの列名を変更する

Question

各DataFramesに適用して列名をクリーンアップできる関数を作成したいとします。例えば

：私は、コード使用して、それを適用するときに

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

しかし、：

df2 = df.apply(standardize_headers) 

を私はエラーを取得：

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

は 'External_Code' 最初の列でありますdfで私はラムダ関数としてこの単純な例を行うことができますが、私は複数のステップでより複雑な関数を持つことを計画していますので、ユーザー定義関数で列名を参照できるようにする必要があります。私はapplyがオブジェクトをシリーズとして扱うことを知っています。だから、私はどのようにユーザー定義関数内の列を参照できますか？または、問題は私が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) 

あなたはどちらか、単に名前を変更したデータフレームを返すために、このテンプレートを展開することができ、またはそれを返す前にデータフレームに適用する複数の機能を渡します。

Answer 2

これを行う簡単な方法の1つはrenameメソッドです。 columnsパラメータに関数を渡すことができ、この関数は各列名に適用されます。関数内のxは文字列であるため、文字列メソッドを直接使用できます。

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

df.rename(columns=standardize_headers) 

apply関数は、各列の値をSeriesとして直接使用します。