numpy入れるとか内包表現使うとかしましょう