Question

我使用以下命令用python写入csv文件：

submission = pandas.DataFrame({
    "label": data["label"],
    predictors[0]: data[predictors[0]]
})

后来我用：

写了一个csv文件

submission.to_csv("test.csv", index=False)

现在，我的预测器列表是1000多个元素的长列表，显然我不能手工编写所有这些行

submission = pandas.DataFrame({
    "label": data["label"],
    predictors[0]: data[predictors[0]]
    predictors[1]: data[predictors[1]]
    predictors[2]: data[predictors[2]]
    predictors[3]: data[predictors[3]]

    ...
    })

我试过

submission = pandas.DataFrame({
    "label": data["label"],
    for i in range(lengthofpredictors):
       predictors[i]: data[predictors[i]]
})

但它不起作用，如

submission = pandas.DataFrame({
    "label": data["label"],
    predictors: data[predictors]
})

不起作用。我该怎么做？

Answer 1

如何使用列表理解来创建字典？

tmp = dict([("label", data["label"])] + \
    [(predictors[i], data[predictors[i]]) for i in range(784)])

然后

 submission = pandas.DataFrame(tmp)

使用pandas编写长列表以保存csv文件

1 个答案: