提取文档字符串中的多个变量名

import re

def extract_variables(docstring):
    pattern = r'@param\s+(\w+)'
    variables = re.findall(pattern, docstring)
    return variables

# 示例文档字符串
docstring = '''
    This function takes two parameters:
    @param x: the first parameter
    @param y: the second parameter
    @return: the sum of x and y
'''

# 提取变量名
variables = extract_variables(docstring)
print(variables)