Экспорт столбца с нулями в CSV - PullRequest
0 голосов
/ 08 мая 2019

Из большого набора данных я пытаюсь экспортировать два столбца в csv: 'row_id' и 'accept'. «Принятый» содержит 0 или 1 на строку. Это результат модели машинного обучения.

Однако, когда я проверяю CSV, все 0 исчезли. Другими словами, если значение «принятого» столбца в моих данных равно 0, в csv оно пустое. Как я могу это исправить?

Я попытался определить тип данных следующим образом:

my_submission=pd.DataFrame({'row_id':testvalues.row_id,'accepted': solution2}).astype('str')

Но вывод был тот же.

Мой текущий код:

solution2=gbr.predict(testvalues1)
my_submission=pd.DataFrame({'row_id':testvalues.row_id,'accepted': solution2})
my_submission.to_csv('GradientBoosterClassifierSolution.csv', index=False)

Текущие результаты:

row_id, accepted
0,1
1
2
3,1
4

Ожидаемые результаты:

row_id, accepted
0,1
1,0
2,0
3,1
4,0

1 Ответ

0 голосов
/ 08 мая 2019

Предполагая, что в вашем списке нет данных, добавьте na_rep к вашему my_submission.to_csv() следующим образом

my_submission.to_csv('GradientBoosterClassifierSolution.csv', index=False, na_rep='0')

Это должно заполнить все пробелы 0

...