WordPressでカテゴリーによって最大表示数を変える

【備忘録】
カテゴリー一覧の最大表示件数をカテゴリーによって変える場合
functions.phpに以下を追記。
tagページの場合は該当部分を置き換え。

function change_pre_get_posts($query){
  if(!is_admin() && $query->is_main_query()){
    // カテゴリーID5は10件
    if(is_category('5')){
      $query->set('posts_per_page', 10);
    }
    // カテゴリーID6は20件
    if(is_category('6')){
      $query->set('posts_per_page', 20);
    }
  }
}
add_action('pre_get_posts', 'change_pre_get_posts');