有没有一种简单的方法可以将熊猫DataFrame上的大字符串拆分成相等数量的单词？

import pandas as pd

# 创建一个包含大字符串的熊猫DataFrame
df = pd.DataFrame({'text': ['This is a long string that needs to be split into equal number of words',
                            'Another long string that should be split into equal number of words']})

# 定义要拆分的单词数量
num_words = 5

# 使用split()函数将大字符串拆分成相等数量的单词
df['split_text'] = df['text'].apply(lambda x: ' '.join(x.split()[:num_words]))

# 打印结果
print(df)

                                                text                    split_text
0  This is a long string that needs to be split i...  This is a long string that
1  Another long string that should be split into ...  Another long string that should