如何使用js或jquery在html表格中点击时突出显示一列？

<table>
  <thead>
    <tr>
      <th>列1</th>
      <th class="highlight-column">列2</th>
      <th>列3</th>
      <th>列4</th>
      <!-- 其他列 -->
    </tr>
  </thead>
  <tbody>
    <!-- 表格数据行 -->
  </tbody>
</table>

$(document).ready(function() {
  $('th.highlight-column').click(function() {
    // 移除其他列的高亮样式
    $('th.highlight-column').removeClass('highlight');
    
    // 添加高亮样式到当前点击的列
    $(this).addClass('highlight');
    
    // 获取列的索引
    var columnIndex = $(this).index();
    
    // 遍历表格的每一行，突出显示对应列的单元格
    $('tbody tr').each(function() {
      $(this).find('td').eq(columnIndex).addClass('highlight');
    });
  });
});

.highlight {
  background-color: yellow;
}