我正在尝试编写一段代码来消除文档中不必要的空格

import re

def remove_extra_spaces(document):
    # 使用正则表达式匹配连续的空格，并替换为空字符串
    cleaned_document = re.sub(r'\s+', '', document)
    return cleaned_document

# 示例文档
document = "这是 一段 有  不必要  空格的 文档。"

# 调用函数消除空格
cleaned_document = remove_extra_spaces(document)
print(cleaned_document)

这是一段有不必要空格的文档。