pandas python:向df添加空白列

时间:2015-06-17 10:55:13

标签: python pandas insert dataframe col

我正在尝试将x个空白列添加到数据框中。

这是我的功能:

def fill_with_bars(df, number=10):
    '''
    Add blank, empty columns to dataframe, at position 0
    '''

    numofcols = len(df.columns)

    while numofcols < number:
        whitespace = ''
        df.insert(0, whitespace, whitespace, allow_duplicates=True)
        whitespace += whitespace
    return df

但我收到此错误

ValueError: Wrong number of items passed 2, placement implies 1

我不确定我做错了什么?

2 个答案:

答案 0 :(得分:2)

我不是一次插入一列,而是创建所需尺寸的df,然后调用concat

In [72]:
def fill_with_bars(df, number=10):
    return pd.concat([pd.DataFrame([],index=df.index, columns=range(10)).fillna(''), df], axis=1)
​
df = pd.DataFrame({'a':np.arange(10), 'b':np.arange(10)})
fill_with_bars(df)

Out[72]:
  0 1 2 3 4 5 6 7 8 9  a  b
0                      0  0
1                      1  1
2                      2  2
3                      3  3
4                      4  4
5                      5  5
6                      6  6
7                      7  7
8                      8  8
9                      9  9

至于为什么会出现这个错误:

这是因为你的str不是一个空格,它是一个空字符串:

In [75]:
whitespace = ''
whitespace + whitespace
Out[75]:
''

所以在第3次迭代中,它试图查找列,期望只有一个列,但是有2个因此它会使内部检查失效,因为它现在找到了2个名为''的列

答案 1 :(得分:1)

试试这个:

def fill_with_bars(old_df, number=10):
    empty_col = [' '*i for i in range(1,number+1)]
    tmp = df(columns=empty_col)
    return pd.concat([tmp,old_df], axis=1).fillna('')