python - Pandas:从特定列标题获取字符串

Question

上下文:

我是 Pandas 新手，我需要一个基于现有列创建新列的函数。新列名称将具有原始列的名称加上新字符(例如:从“As”列创建“As NEW”列)。我可以访问旧的列标题字符串来创建新列的名称吗？

问题:

我有df['columnA']并且需要获取“columnA”字符串

Answer 1

如果我理解正确的话，这可能就是您正在寻找的。

您可以使用str.contains()对于列，然后使用 string formatting创建新的列名称。

df = pd.DataFrame({'col1':['A', 'A', 'B','B'], 'As': ['B','B','C','C'], 'col2': ['C','C','A','A'], 'col3': [30,10,14,91]})
col = df.columns[df.columns.str.contains('As')]
df['%s New' % col[0]] = 'foo'
print (df)
   As   col1 col2  col3 As New
0  B    A    C    30    foo
1  B    A    C    10    foo
2  C    B    A    14    foo
3  C    B    A    91    foo