import pickle
import numpy as np
import pandas as pd

predict = [1,2,3,4]

df = pd.DataFrame([(f"testid{i:04}", v) for i, v in enumerate(predict)], columns=["id", "pred"])
df.to_csv("submission.csv", index=None)