如何根据单词的长度替换单词？

def replace_words_by_length(text, target_length, new_word):
    words = text.split(" ")
    replaced_words = []
    for word in words:
        if len(word) == target_length:
            replaced_words.append(new_word)
        else:
            replaced_words.append(word)
    replaced_text = " ".join(replaced_words)
    return replaced_text

text = "This is a sample text for word replacement."
target_length = 5
new_word = "REPLACED"
replaced_text = replace_words_by_length(text, target_length, new_word)
print(replaced_text)

This is a REPLACED text for word REPLACED.