Использование Pandas str.split с различной длиной строк через запятую

Question

Я пытаюсь использовать:

ball_inf = ball_inf[['x','y','z', 'speed', 'Ball_Ownership', 'Ball_InPlay']] = ball_inf['base'].str.split('-',expand=True)

, чтобы заполнить кадр данных одним столбцом, называемым «base», который содержит разделенные запятыми строки различной длины 6 или 7, то есть

"28,-7,0,82.00,A,Dead"
"38,-5,0,83.00,A,Dead,Go"

Я получаю предупреждение:

Columns must be same length as key

Можно ли добавить NA, если нет 7-го предмета?

Answer 1

Конечно.Просто используйте str.split с , и join с исходным кадром данных:

df = pd.DataFrame({'strings': ['28,-7,0,82.00,A,Dead', '38,-5,0,83.00,A,Dead,Go']})

df = df.join(df.pop('strings').str.split(',', expand=True))

print(df)

    0   1  2      3  4     5     6
0  28  -7  0  82.00  A  Dead  None
1  38  -5  0  83.00  A  Dead    Go