我正在尝试创建一个包含2列的新数据框。第一列为id
,第二列为predictions
。我想创建一个第一列id
的数据框,该列取自testdata
而predictions
是一个0's
和1's
的数组。列名应为id
和predictions
。
示例数据:
我的新创建数据框应该是这样的:
预期产出:
id predictions
3242 0
3246 1
7655 1
1324 0
4643 1
我尝试了以下代码但是我收到了一些错误,请查看截图:
答案 0 :(得分:2)
如果testdata
的长度与predictions
相同,我认为需要:
测试它:
print (len(testdata))
print (len(predictions))
df = pd.DataFrame({'id':testdata['id'], 'predictions':predictions})
答案 1 :(得分:1)
如果数组的长度相等
df = pd.DataFrame({'id':testdata['id']})
df["Predictions"] = predictions