Как создать новый столбец из данных в пандах? - PullRequest
0 голосов
/ 28 мая 2018

Я пытаюсь создать новый кадр данных из 2 столбцов.Первый столбец - id, а второй - predictions.Я хочу создать фрейм данных с первым столбцом id, который взят из testdata, а predictions - это массив 0's и 1's.Имена столбцов должны быть id и predictions.

Пример данных:

enter image description here

Мой новый кадр данных для создания должен выглядеть следующим образом:

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

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

Я попробовал следующий код, но получаю ошибку, пожалуйста, проверьте скриншот:

enter image description here

Ответы [ 2 ]

0 голосов
/ 28 мая 2018

Если длина массивов равна

df = pd.DataFrame({'id':testdata['id']})
df["Predictions"] = predictions
0 голосов
/ 28 мая 2018

Я считаю нужным, если длина testdata совпадает с predictions:

Проверьте это:

print (len(testdata))

print (len(predictions))

df = pd.DataFrame({'id':testdata['id'], 'predictions':predictions})
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...