Pandas：向现有数据框添加新的未命名列

Question

我有一个现有的数据帧和一个像这样的np数组：

[[ 0.3397825   0.6602175 ]
 [ 0.3397825   0.6602175 ]
 ...,
 [ 0.89700502  0.10299498]]

此数组中的行数与我的数据帧行数相匹配。我只想在我的数据框中添加两个新列，其列标题为“prob0，prob1”。

我该怎么做？

Answer 1

使用concat

pd.concat([df,pd.DataFrame(ary,columns=['prob0', 'prob1'])],axis=1)

数据输入

ary=[[ 0.3397825 , 0.6602175 ],
 [ 0.3397825 ,  0.6602175 ],
 [ 0.89700502  ,0.10299498]]