一种用于递归迭代器的 iterator_to_array 以获得二维数组

Question

在 PHP 中使用迭代器时，您可以使用 iterator_to_array 函数来提取迭代结果的数组。例如，假设您关注ArrayObject：

$array_object = new ArrayObject(array(
   array('1', '2', '3', '4'),
   array('5', '6', '7', '8'),
   array('9', '10', '11', '12'),
));

如你所见，它的存储是一个二维数组。

我们可以创建一个FilterOperator 来只接受它的第一项（我知道使用LimitIterator 会更好，这只是一个示例目的）：

class myFilterIterator extends FilterIterator
{
   public function accept()
   {
      return ($this->key() === 0);
   }
}

$filter_iterator = new myFilterIterator(new ArrayIterator($array_object));

现在，如果我这样做：

print_r(iterator_to_array($filter_iterator));

如果我手动循环遍历运算符，我会得到数组：

Array ( [0] => Array ( [0] => 1 [1] => 2 [2] => 3 [3] => 4 ) )

但是现在如果我想使用RecursiveFilterIterator 怎么办？假设我有：

class myRecursiveFilterIterator extends RecursiveFilterIterator
{
   public function accept()
   {
      return ($this->hasChildren() || $this->key() === 0);
   }
}

$recursive_filter_iterator = new myRecursiveFilterIterator(new RecursiveArrayIterator($array_object));

如您所见，这将只接受父数组中包含的每个数组的键 0。因此，如果我对其进行递归迭代，它就会起作用：

foreach (new RecursiveIteratorIterator($recursive_filter_iterator) as $value) {
   print_r($value);
   echo '<br />';
}

结果：

1
5
9

但是，我怎样才能快速得到数组array(array(1), array(5), array(9))？

如果我这样做：

print_r(iterator_to_array($recursive_filter_iterator));

或

print_r(iterator_to_array($recursive_filter_iterator->getInnerIterator()));

或

$it = new RecursiveIteratorIterator($recursive_filter_iterator);
print_r(iterator_to_array($it->getInnerIterator()));

我得到了整个原始数组：

Array ( [0] => Array ( [0] => 1 [1] => 2 [2] => 3 [3] => 4 ) [1] => Array ( [0] => 5 [1] => 6 [2] => 7 [3] => 8 ) [2] => Array ( [0] => 9 [1] => 10 [2] => 11 [3] => 12 ) )

如果我这样做：

print_r(iterator_to_array(new RecursiveIteratorIterator($recursive_filter_iterator)));

我只得到第一项：

Array ( [0] => 9 )

如果我这样做：

print_r(iterator_to_array(new RecursiveIteratorIterator($recursive_filter_iterator->getInnerIterator())));

我得到了父数组中的最后一项，但键为 0：

Array ( [0] => 9 [1] => 10 [2] => 11 [3] => 12 ) 

我需要的是获取数组：

Array ( [0] => Array ( [0] => 1 ) [1] => Array ( [0] => 5 ) [2] => Array ( [0] => 9 ) ) 

我知道我可以让它手动循环，但我想知道是否有直接的方法，比如在iterator_to_array 中用于非递归迭代器。当然，我对 PHP 中的递归迭代器有些不理解，但它的文档在这方面真的很糟糕。

非常感谢。

Answer 1

尚不完全清楚您真正想要做什么，但以下采用 RecursiveArrayIterator（注意：ArrayObject 不是 递归迭代器）并使用 iterator_to_array() 来获取你想要的结果数组。

class FirstOnlyRecursiveArrayIterator extends ParentIterator {
    public function __construct(RecursiveArrayIterator $it) {
         parent::__construct($it);
    }
    public function current() {
        $children = parent::current();
        return array_slice($children, 0, 1);
    }
}

$array_it = new RecursiveArrayIterator(array(
    array('1', '2', '3', '4'),
    array('5', '6', '7', '8'),
    array('9', '10', '11', '12'),
));

$filter_iterator = new RecursiveIteratorIterator(
    new FirstOnlyRecursiveArrayIterator($array_it),
    RecursiveIteratorIterator::SELF_FIRST);
print_r(iterator_to_array($filter_iterator));

Answer 2

您需要/想要使用迭代器吗？您可以使用 php 的内置 array_map() 函数，该函数接受一个数组和一个函数，并将指定的函数应用于数组中的每个元素。所以你可以这样做：

<?php
function get_first($foo)
{ 
    return array_slice($foo, 0, 1); //slice the array right after the first element
}

$array_object = new ArrayObject(array(
     array('1', '2', '3', '4'),
     array('5', '6', '7', '8'),
     array('9', '10', '11', '12'),
));

$new_array = array_map("get_first", $array_object->getArrayCopy());
print_r($new_array);
?>

结果将是：

Array
(
    [0] => Array
        (
            [0] => 1
        )

    [1] => Array
        (
            [0] => 5
        )

    [2] => Array
        (
            [0] => 9
        )
)

如果我误解了您的问题，我深表歉意。

Answer 3

好的。一种简单的方法可能是以下。它扩展了 ArrayIterator 类并覆盖 current() 函数以返回切片数组：

<?php
class FirstOnlyIterator extends ArrayIterator
{
    public function current()
    {
            $next = parent::current();
            return array_slice($next, 0, 1);
    }
}

$array_object = new ArrayObject(array(
     array('1', '2', '3', '4'),
     array('5', '6', '7', '8'),
     array('9', '10', '11', '12'),
));

$iterator = new FirstOnlyIterator($array_object);

print_r(iterator_to_array($iterator));
?>