2017-07-22 10 views
1

を作成しますforループは、リストから列名を取得し、リストの要素の一部を含む新しい列名を作成します。各反復では、私たちは、私がパンダで、次の列名を持っているとしましょうループ内の文字列のリストから新しい列名

私は次のよう作成したいと思います:名前A_CとB_c​​を持つ2つの新しい列の追加を生じるはずである

a = ["A", "B"] 
for elements i in a: 
    elements + "_c" = SOME FUNCTION column <df.elements> 

を。

答えて

2

あなたは、ループ内でSeries.applyを使用することができます。その後、新しいdfためadd_suffixjoin

a = ["A", "B"] 
for i in a: 
    df[i + "_c"] = df[i].apply(SOME FUNCTION) 

またはDataFrame.applyを元に:

df1 = df[a].apply(SOME FUNCTION).add_suffix('_c') 
df = df.join(df1) 

サンプル:

df = pd.DataFrame({'A':[4,5,4,5,5,4], 
        'B':[7,8,9,4,2,3], 
        'C':[1,3,5,7,1,0]}) 

print (df) 
    A B C 
0 4 7 1 
1 5 8 3 
2 4 9 5 
3 5 4 7 
4 5 2 1 
5 4 3 0 

def FUNCTION(x): 
    return x + 10 

a = ["A", "B"] 

for i in a: 
    df[i + "_c"] = df[i].apply(FUNCTION) 

print (df) 
    A B C A_c B_c 
0 4 7 1 14 17 
1 5 8 3 15 18 
2 4 9 5 14 19 
3 5 4 7 15 14 
4 5 2 1 15 12 
5 4 3 0 14 13 

df = df.join(df[a].apply(FUNCTION).add_suffix('_c')) 
print (df) 
    A B C A_c B_c 
0 4 7 1 14 17 
1 5 8 3 15 18 
2 4 9 5 14 19 
3 5 4 7 15 14 
4 5 2 1 15 12 
5 4 3 0 14 13 
+0

ありがとうございました!それはうまく動作します。私の問題は、カラムの名前を含む名前を持つFUNCTIONの中にパラメータを指定しなければならないということでした。 – Guga

+0

それは何が必要なのかによって異なります - 関数で直列に動作する必要がある場合は 'x'を使用し、必要な場合は関数' FUNC'で 'x.name'を使用します。 – jezrael

関連する問題