如何从pandas中的给定数据创建新列?

时间:2018-05-28 06:52:38

标签: python pandas

我正在尝试创建一个包含2列的新数据框。第一列为id,第二列为predictions。我想创建一个第一列id的数据框,该列取自testdatapredictions是一个0's1's的数组。列名应为idpredictions

示例数据:

enter image description here

我的新创建数据框应该是这样的:

预期产出:

id      predictions
3242        0
3246        1
7655        1
1324        0
4643        1

我尝试了以下代码但是我收到了一些错误,请查看截图:

enter image description here

2 个答案:

答案 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