import pandas as pd test_df = pd.read_csv("/tmp/data/test.csv") ids = test_df['ID'].values submit_df = pd.DataFrame([(id, 1) for id in ids], columns=["id", "pred"]) submit_df.to_csv('submission.csv', index=None)