Скрытие столбцов в таблице

Таблица

<table border="1" id="table">
  <tr>
   <td>0</td>
   <td>1</td>
   <td>2</td>
  </tr>
  <tr>
   <td>0</td>
   <td>1</td>
   <td>2</td>
  </tr>                            
</table>

при нажатие на checkbox скрывается столбец

<input name="column" type="checkbox" value="0" onClick="toggle_column('table', this.value);" checked />
<input name="column" type="checkbox" value="1" onClick="toggle_column('table', this.value);" checked />
<input name="column" type="checkbox" value="2" onClick="toggle_column('table', this.value);" checked />

scrypt

 function toggle_column(table, column_nr) {
        table = document.getElementById('table');
        rows = table.getElementsByTagName('tr');
        for (var row=0; row<rows.length;row++) {
            cels = rows[row].getElementsByTagName('td');
            cels[column_nr].style.display=cels[column_nr].style.display=="none" ? "" : "none";
        }
    }

Можно ли как то сделать что бы скрывались столбцы не по value, а по id или class, хочу применить к таблице такого вида

<table  class="table table-bordered table-hover">
     <thead>
      <tr>
        <th><?php echo $text_price ?></th>
        <th><?php echo $text_comment?></th>
       </tr>
      </thead>
      <tbody>
      <?php foreach ($orders as $order) { ?>
        <tr>
          <td><?php echo $order['comment']?></td>
        </tr>
       <?php } ?>
      </tbody>
  </table>