用PHP关联数组求笛卡儿积

Question

假设我有一个数组，如下所示：Array(     [arm] => Array         (             [0] => A            [1] => B            [2] => C        )     [gender] => Array         (             [0] => Female             [1] => Male         )     [location] => Array         (             [0] => Vancouver             [1] => Calgary         ))如何在保留外部关联数组的键并在内部数组中使用它们的同时，找到笛卡儿积？算法的结果应该是：Array(     [0] => Array         (             [arm] => A            [gender] => Female             [location] => Vancouver         )     [1] => Array         (             [arm] => A            [gender] => Female             [location] => Calgary         )     [2] => Array         (             [arm] => A            [gender] => Male             [location] => Vancouver         )...etc.我已经查阅了很多笛卡尔积算法，但我仍然停留在如何保存关联键的细节上。我目前使用的算法只给出数字索引：    $result = array();     foreach ($map as $a) {         if (empty($result)) {             $result = $a;             continue;         }         $res = array();         foreach ($result as $r) {             foreach ($a as $v) {                 $res[] = array_merge((array)$r, (array)$v);             }         }         $result = $res;     }     print_r($result);任何帮助都将不胜感激。用PHP关联数组求笛卡儿积

Answer 1

我能想到的是：

function inject($elem, $array) {
    return array_map(function ($n) use ($elem) { return array_merge((array)$elem, (array)$n); }, $array);}function zip($array1, $array2) {
    return array_reduce($array1, function ($v, $n) use ($array2) { return array_merge($v, inject($n, $array2));  }, array());}
    function cartesian_product($array) {
    $keys = array_keys($array);
    $prod = array_shift($array);
    $prod = array_reduce($array, 'zip', $prod);
    return array_map(function ($n) use ($keys) { return array_combine($keys, $n); }, $prod);}

---

(下面使用伪数组/列表/字典表示法，因为PHP对于这类事情太冗长了。)

这个inject函数变换a, [b]进[(a,b)]，即它向数组的每个值注入一个值，返回一个数组。不管是不是a或b已经是一个数组了，它将始终返回一个二维数组。

inject('a', ['foo', 'bar'])
    =>  [('a', 'foo'), ('b', 'bar')]

这个zip函数应用inject函数到数组中的每个元素。

zip(['a', 'b'], ['foo', 'bar'])
    =>  [('a', 'foo'), ('a', 'bar'), ('b', 'foo'), ('b', 'bar')]

请注意，这实际上产生了笛卡尔积，因此zip有点用词不当。简单地将此函数应用于数据集中的所有元素，就可以得到任意长度数组的笛卡儿积。

zip(zip(['a', 'b'], ['foo', 'bar']), ['42', '76'])
    =>  [('a', 'foo', '42'), ('a', 'foo', '76'), ('a', 'bar', '42'), …]

这不包含键，但是由于元素在结果集中都是按顺序排列的，所以您可以简单地将键重新注入结果。

array_combine(['key1', 'key2', 'key3'], ['a', 'foo', '42'])
    =>  [ key1 : 'a', key2 : 'foo', key3 : '42' ]

将其应用于产品中的所有元素，将得到所需的结果。

如果您愿意，可以将上述三个函数折叠成一个长语句(这也可以清除错误的名称)。

---

对于PHP<=5.2没有匿名函数的“展开”版本如下所示：

function inject($elem, $array) {
    $elem = (array)$elem;
    foreach ($array as &$a) {
        $a = array_merge($elem, (array)$a);
    }
    return $array;}function zip($array1, $array2) {
    $prod = array();
    foreach ($array1 as $a) {
        $prod = array_merge($prod, inject($a, $array2));
    }
    return $prod;}function cartesian_product($array) {
    $keys = array_keys($array);
    $prod = array_shift($array);
    $prod = array_reduce($array, 'zip', $prod);

    foreach ($prod as &$a) {
        $a = array_combine($keys, $a);
    }
    return $prod;}