費用関数の方もpylabを使わないよう変更しました。