Увеличьте длину столбца Dataframe до определенного количества слов, добавив

Question

Мой фрейм данных выглядит как

      Abc                       XYZ 
0  Hello      How are you doing today
1   Good                    This is a
2    Bye                      See you
3  Books     Read chapter 1 to 5 only

max_words = 6, filler_word = 'end'. В столбце XYZ я хочу добавить его так, чтобы все строки имели длину max_words.

Требуемый вывод

     Abc                       XYZ
0  Hello               How are you end end end
1   Good               This is a end end end
2    Bye               See you end end end end
3  Books               Read chapter 1 to 5 only

Строка 3 не заполнена, поскольку ее длина уже равна 6.

Answer 1

IIU C, попробуйте это:

df['XYZ'] = df['XYZ'].str.split(expand=True)\
                     .fillna('end')\
                     .apply(lambda x: x.str.cat(sep=' '), axis=1)

print(df)

Вывод:

     Abc                          XYZ
0  Hello  How are you doing today end
1   Good        This is a end end end
2    Bye      See you end end end end
3  Books     Read chapter 1 to 5 only