如何对表头上的表格(单击)进行排序？

<table id="myTable">
  <thead>
    <tr>
      <th onclick="sortTable(0)">姓名</th>
      <th onclick="sortTable(1)">年龄</th>
      <th onclick="sortTable(2)">成绩</th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <td>张三</td>
      <td>25</td>
      <td>80</td>
    </tr>
    <tr>
      <td>李四</td>
      <td>22</td>
      <td>90</td>
    </tr>
    <tr>
      <td>王五</td>
      <td>28</td>
      <td>70</td>
    </tr>
  </tbody>
</table>

<script>
function sortTable(columnIndex) {
  var table = document.getElementById("myTable");
  var rows = Array.from(table.rows);
  var isAscending = true; // 默认升序排序

  rows.sort(function(row1, row2) {
    var cell1 = row1.cells[columnIndex].textContent;
    var cell2 = row2.cells[columnIndex].textContent;

    if (columnIndex === 0 || columnIndex === 2) {
      // 字符串比较
      return cell1.localeCompare(cell2);
    } else {
      // 数字比较
      return parseInt(cell1) - parseInt(cell2);
    }
  });

  if (!isAscending) {
    rows.reverse(); // 降序排序
  }

  rows.forEach(function(row) {
    table.appendChild(row); // 重新插入排序后的行数据
  });

  isAscending = !isAscending; // 切换排序顺序
}
</script>