在Jupyter Python中突出显示相同的变量

from IPython.display import display, HTML

def highlight_variable(var):
    js_code = f"""
    <script type="text/javascript">
        var cells = document.getElementsByClassName('jp-CodeCell-inputWrapper');
        for (let i = 0; i < cells.length; i++) {{
            let cell = cells[i];
            let text = cell.innerText;
            let regex = new RegExp("\\\\b" + "{var}" + "\\\\b", "g");
            cell.innerHTML = text.replace(regex, "<span style='background-color: yellow;'>{var}</span>");
        }}
    </script>
    """
    display(HTML(js_code))

# 输入你要突出显示的变量名
highlight_variable('variable_name')