递归迭代多维数组，返回相同的数组结构，并在PHP中插入新的键/值

Question

我试图编写一个代码片段，该代码段采用多维数组，并在找到命名搜索键的同一级别插入一些键。我不必依赖于数组的结构(但将在最多5个级别)我不能使用引用传递，所以传统的循环函数对这种方法没有帮助。

我有两个选择:SPL或递归，重新构建数组并一路上改变它

与SPL我似乎不能插入一个新的值..

            $a= new ArrayObject($priceConfig);
            $array = new RecursiveArrayIterator($a);
            $iterator = new RecursiveIteratorIterator($array, RecursiveIteratorIterator::SELF_FIRST);
            foreach ($iterator as $key => $value) {
                if (is_array($value) && $key == 'prices') {
                    $iterator->offsetSet('myPrice',['amount'=>'1.00']);
                }
            }
            print_r($a->getArrayCopy());

它不会在所需的级别插入新键，但它会循环遍历数组。我错过了什么?

重建数组并在嵌套数组中的键搜索处插入新值的递归函数可以工作，但我想使用迭代器来完成此操作…

             function recursive( $input, $searchKey, $key=null) {
                $holder = array();
                if(is_array( $input)) {
                    foreach( $input as $key => $el) {
                        if (is_array($el)) {
                            $holder[$key] = recursive($el, $searchKey, $key);
                            if ($key == $searchKey) {
                                $holder[$key]['inertedPrice'] = "value";
                            }
                        } else {
                            $holder[$key] = $el;
                        }
                    }
                }
                return $holder;
            }

INPUT(总是会有一些"价格键和X水平的结构")

    [1] => Array
        (
            [1] => Array
                (
                    [prices] => Array
                        (
                            [onePrice] => Array( [amount] => 10)
                            [finalPrice] => Array ([amount] => 10)
                        )
                    [key1] => value2
                    [key2] => value2
                )
            [2] => Array
                (
                    [prices] => Array
                        (
                            [otherPrice] => Array([amount] => 20)
                            [finalPrice] => Array([amount] => 20)
                        )
                    [key] => value
                )
        )
)

输出

[1] => Array
    (
        [1] => Array
            (
                [prices] => Array
                    (
                        [onePrice] => Array( [amount] => 10)
                        [finalPrice] => Array ([amount] => 10)
                        [INSERTEDPrice] => Array([amount] => value)
                    )
                [key1] => value2
                [key2] => value2
            )
        [2] => Array
            (
                [prices] => Array
                    (
                        [otherPrice] => Array([amount] => 20)
                        [finalPrice] => Array([amount] => 20)
                        [INSERTEDPrice] => Array([amount] => )
                    )
                [key] => value
            )
    )

)

Answer 1

您可以通过使用自定义迭代器逻辑扩展RecursiveArrayIterator来使用迭代器:

class Foo extends RecursiveArrayIterator
{
    public function getChildren()
    {
        if ($this->key() == 'prices') {
            return new self(array_merge($this->current(), ['foo' => 'bar']));
        } else {
            return parent::getChildren();
        }
    }
}

Answer 2

您可以使用foreach循环和递归等基本工具相当容易地解决这个问题。这就是这样一个解决方案。

function mergeWithKey($targetKey, $new, array $array) {
  foreach ($array as $key => $value) {
    if ($key === $targetKey) {
      $array[$key] = array_merge($array[$key], $new);
    }
    elseif (is_array($value)) {
      $array[$key] = mergeWithKey($targetKey, $new, $value); 
    }
  }
  return $array;
}
// Example
$output = mergeWithKey('prices', array('INSERTEDPrice' => 'value'), $input);

简单地说，当我们遍历数组时，如果我们找到了我们要找的键，那么我们就合并新的价格。如果我们找到一个子数组，那么我们将新的价格合并到这个子数组中。

我通过保留键"prices"作为参数来一般化这个函数。这可能仍然是一个不太可能被重用的不稳定函数。

使用一些常见的算法，您可以很好地构建这个函数，并且可以重用这些算法。一种是同时映射键和值的数组，另一种是将2D数组扁平化为1D数组。

function arrayMapWithKey(callable $f, array $array) {
  $out = array();
  foreach ($array as $key => $value) {
    $out[$key] = $f($key, $value);
  }
  return $out;
}
function concat(array $array) {
  if (empty($array)) {
    return array(); 
  }
  else {
    return call_user_func_array('array_merge', $array);
  }
}

这些定义使您能够编写替代解决方案。

function addPrice($name, $price, $data) {
  return concat(arrayMapWithKey(
    function ($k, $v) use ($name, $price) {
      if ($k === 'prices') {
        return array($k => array_merge($v, array($name => $price)));
      }
      elseif (is_array($v)) {
        return array($k => addPrice($name, $price, $v));  
      }
      else {
        return array($k => $v);
      }
    },
    $data
  ));
}

arrayMapWithKey的另一个公式是用值复制键，然后使用常规的array_map。

function arrayWithKeys(array $array) {
  $out = array();
  foreach ($array as $key => $value) {
    // in PHP arrays are often used as tuples,
    // and here we have a 2-tuple.
    $out[] = array($key, $value);
  }
  return $out;
}