回答編集履歴

- CSV ファイルからクラスの情報もインデックスではなく列（列名: `'class'`）として読みみ込み，それを用いて必要なデータを抽出するのもひとつの方法かもしれません。