将一组标头添加到现有数据框

Posted 2021-04-10

我需要将一组附加标题（作为空白列）附加到现有数据框;相应的值将在以后的过程中添加。请指教，谢谢！

答案

这是一种方式，我已经包含了一些示例数据。

import pandas as pd
df = pd.DataFrame({'c1': [1, 2, 3],
                   'c2': [1, 1, 1],
                   'c3': ['a', 'b', 'c'],
                   'c4': ['a', 'a', 'a'],
                   'cat': ['x', 'x', 'y']})

# These are the column headers you want to add
cols_to_add = ['a', 'b', 'c', 'd', 'e']

码：

df = df.assign(**dict((col, '') for col in cols_to_add))

输出：

   c1  c2 c3 c4 cat a b c d e
0   1   1  a  a   x          
1   2   1  b  a   x          
2   3   1  c  a   y          

说明：DataFrame.assign允许您将列分配给DataFrame。您可以将解压缩的字典作为关键字参数传递给一次分配多个列，在这种情况下，列名由列表中的名称给出，值为空字符串。

另一答案

您可以将pd.DataFrame.join与包含空列的数据框一起使用：

cols_to_add = list('abcde')
df = df.join(pd.DataFrame(columns=cols_to_add))