PHP

PHPで array_find_index() をしてみる

PHPでも array.findIndex() をやりたい。
array_search() では検索条件が指定できないのがもどかしい。
array_map() と array_filter() と array_reduce() はあるのに、どうしてかこれは無いようなので作ってみた。（私が知らないだけで本当は違う名前の関数があるんだろうか・・車輪の再発明でないことを祈る）
最初に見つけた要素の添字が返る。
見つからなかったときは -1 が返る。

function array_find_index($arr, $callback) { $r = array_filter($arr, $callback); return count($r) ? array_keys($r)[0] : -1; } // 使い方 $ar = [ ['key1' => 'hoge', 'key2' => 'fuga', 'key3' => 'piyo'], ['key1' => 'foo', 'key2' => 'bar', 'key3' => 'buz'], ['key1' => 'dog', 'key2' => 'cat', 'key3' => 'mouse'], ]; $res = array_find_index($ar, fn($a) => $a['key2'] === 'cat'); print_r($res);

ついでに array_find() も。
最初に見つけた要素そのものが返る。
見つからなかったときは NULL が返る。

function array_find($arr, $callback) { $r = array_filter($arr, $callback); return count($r) ? array_shift($r) : NULL; } // 使い方 $ar = [ ['key1' => 'hoge', 'key2' => 'fuga', 'key3' => 'piyo'], ['key1' => 'foo', 'key2' => 'bar', 'key3' => 'buz'], ['key1' => 'dog', 'key2' => 'cat', 'key3' => 'mouse'], ]; $res = array_find($ar, fn($a) => $a['key2'] === 'cat'); print_r($res);

条件式に合う要素が存在するかどうかだけ知りたいのであれば、もう少し処理を削れる。
コールバック関数が使える in_array() のような array_contains() 関数も作ってみた。
ただし結果は int で返る。（つまり条件に合う要素が何個含まれるのか、が分かる）

function array_contains($arr, $callback) { return count(array_filter($arr, $callback)); } // 使い方 $ar = [ ['key1' => 'hoge', 'key2' => 'fuga', 'key3' => 'piyo'], ['key1' => 'foo', 'key2' => 'bar', 'key3' => 'buz'], ['key1' => 'dog', 'key2' => 'cat', 'key3' => 'mouse'], ]; $res = array_contains($ar, fn($a) => $a['key2'] === 'cat'); if ($res) { print "found!\n"; } else { print "not found...\n"; }

後記：PHP8.4で正式にarray_findが追加されましたね・・。
諸事情あって古いPHPを使っている方はどうぞ。