PHP | Function to assign a value to any multidimensional array with array of keys

Question

My goal is to write a function which would assign a value to any multidimensional array:

function array_multidim_set($arr, $keys, $value)

Example: print_r(array_multidim_set($arr, ['key1', 'key2', 'key'3'], 'foo') should create a value as following $arr[key1][key2][key3] = 'foo';

For now I'm using this:

function array_multidim_set($arr, $keys, $value){
    switch(count($keys)){
        case 1:
           $arr[$keys[0]] = $value;
           break;
        case 2:
           $arr[$keys[0]][$keys[1]] = $value;
           break;
        case 3:
           $arr[$keys[0]][$keys[1]][$keys[2]] = $value;
           break;

        ...and so on...

    }

    return $arr;
}

But it's limited to the amount of cases defined. Is there any way to make a universal function for any amount of keys?

Thank you!

Chetan Ameta · Accepted Answer · 2017-10-24 09:20:52Z

3

try below solution:

function array_multidim_set(&$arr, $keys, $value){
    $rv = &$arr;
    foreach($keys as $pk)
    {
        $rv = &$rv[$pk]; // Unused reference [ex. $rv['key1'] then $rv['key1']['key2'] .. so on ] - actually assigned to $target by reference
    }
    $rv = $value;
}

$target = ['test' => 'test'];

array_multidim_set($target, ['key1', 'key2', 'key3'], 'foo');
echo '<pre>';
print_r($target);

Output:

Array
(
    [test] => ttt
    [key1] => Array
        (
            [key2] => Array
                (
                    [key3] => foo
                )

        )

)

answered Oct 24, 2017 at 9:20

Chetan Ameta

7,8943 gold badges34 silver badges46 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Dainius Vaičiulis Over a year ago

That is really awesome! Thanks!

Chetan Ameta Over a year ago

Thanks @DainiusVaičiulis

Collectives™ on Stack Overflow

PHP | Function to assign a value to any multidimensional array with array of keys

1 Answer 1

2 Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Related