假设我在 pandas 中有以下列名称: [“A”,“B”]
我的问题是我想使用 for 循环从列表中获取列名称并创建一个新的列名称,其中包含列表中的部分元素。
在每次迭代中我想创建以下内容:
a = ["A", "B"]
for elements i in a:
elements + "_c" = SOME FUNCTION column <df.elements>
这应该会导致添加两个名为 A_c 和 B_c 的新列。
最佳答案
您可以使用Series.apply
循环中:
a = ["A", "B"]
for i in a:
df[i + "_c"] = df[i].apply(SOME FUNCTION)
或者DataFrame.apply
与 add_suffix
对于新的 df
然后 join
原文:
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
关于python - 从循环中的字符串列表创建新的列名称,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45250425/