JavaScript: [🐈, 🐕, 🐕, 🐈, 🐈, 🐕]のような配列を[[🐈, 🐈, 🐈 ], [🐕, 🐕, 🐕]]に仕分けする

JavaScriptの配列を要素の内容に応じて、2つのグループに仕分け(partitioning)する方法です。

下の例では、要素が犬か猫かによって仕分けしています:

partition.js

const pets = ['🐈', '🐕', '🐕', '🐈', '🐈', '🐕']

const [cats, dogs] = pets.reduce(
  ([cats, dogs], pet) =>
    pet === '🐈' ? [[...cats, pet], dogs] : [cats, [...dogs, pet]],
  [[], []]
)

console.log(cats, dogs)
//=> [ '🐈', '🐈', '🐈' ] [ '🐕', '🐕', '🐕' ]

ちなみにTypeScriptではreduceの型引数に[string[], string[]]を指定します:

partition.ts

const pets = ['🐈', '🐕', '🐕', '🐈', '🐈', '🐕']

const [cats, dogs] = pets.reduce<[string[], string[]]>(
  ([cats, dogs], pet) =>
    pet === '🐈' ? [[...cats, pet], dogs] : [cats, [...dogs, pet]],
  [[], []]
)

console.log(cats, dogs)

仕訳処理を汎用化すると:

partition.js

const partition = (array, predicate) =>
  array.reduce(
    ([left, right], value) =>
      predicate(value) ? [[...left, value], right] : [left, [...right, value]],
    [[], []]
  )
const pets = ['🐈', '🐕', '🐕', '🐈', '🐈', '🐕']
const [cats, dogs] = partition(pets, value => value === '🐈')

partition.ts

const partition = <
  T extends ReadonlyArray<any>,
  U extends T[number],
  V extends Exclude<T[number], U>
>(
  array: T,
  predicate: (value: T[number]) => value is U
): [U[], V[]] =>
  array.reduce(
    ([left, right], value) =>
      predicate(value) ? [[...left, value], right] : [left, [...right, value]],
    [[], []]
  )

const pets = ['🐈', '🐕', '🐕', '🐈', '🐈', '🐕'] as const
const [cats, dogs]: ['🐈'[], '🐕'[]] = partition(pets, (value): value is '🐈' => value === '🐈')

---

最後までお読みくださりありがとうございました。Twitterでは、Qiitaに書かない技術ネタなどもツイートしているので、よかったらフォローしてもらえると嬉しいです→Twitter@suin