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

Question

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

Answer 1

您可以对包含空列的数据框使用pd.DataFrame.join：

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

Answer 2

这是一种方式，我提供了一些示例数据。

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。您可以将其作为关键字参数传递给未包装的字典，以一次分配多个列，在这种情况下，列名由列表中的名称指定，而值作为空字符串。