Get new array with specific columns from existing 2d array

extract_new_array.php

$api_records = getDataFromAPI(); // Fetchh data from DB/ or from another API

$keys = array('productID' => '', 'code' => '', 'parentProductID' => '');
$product_ids = getArrayFrom2dArray($api_records, $keys);


/**
 *
 * @param array $records [2 dimensional associative array]
 * @param array $keys
 * @return array
 *
 * $keys must be array key only.
 * $keys = array('productID' => '', 'code' => '', 'parentProductID' => '');
 */
function getArrayFrom2dArray(array $records, $keys) {
    return array_map(
        function($arr) use ($keys)  {
            if (is_array($keys)) {
                // This will only return the all those keys=>values which are exits in source array
                return array_intersect_key($arr, $keys);
            }
            else {
                // This will always return the parent_product_id in return array
                // Irrespctive of that key is present in source array or not
                $parent_id = (isset($arr['parentProductID'])) ? $arr['parentProductID'] : 0;
                return array('productID' => $arr['productID'], 'sku' => $arr['code'], 'parentProductID' => $parent_id);
            }
        }, $records
    );
}