限制 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_size = 3, 我想将列（XYZ）截断为最大 3 个字（max_size）。有长度小于max_size的行，应该保持原样。

期望的输出：

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

Answer 1

使用带限制的拆分，删除最后一个值，然后将列表连接在一起：

max_size = 3

df['XYZ'] = df['XYZ'].str.split(n=max_size).str[:max_size].str.join(' ')
print (df)
     Abc             XYZ
0  Hello     How are you
1   Good       This is a
2    Bye         See you
3  Books  Read chapter 1

使用 lambda 函数的另一种解决方案：

df['XYZ'] = df['XYZ'].apply(lambda x: ' '.join(x.split(maxsplit=max_size)[:max_size]))