熊貓：將新的未命名列添加到現有數據框

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]]