2016-01-20 64 views
0
CREATE OR REPLACE FUNCTION array_replace(INT[]) RETURNS float[] AS $$ 
DECLARE 
    arrFloats ALIAS FOR $1; 
    J int=0; 
    x int[]=ARRAY[2,4]; 
    -- xx float[]=ARRAY[2.22,4.33]; 
    b float=2.22; 
    c float=3.33; 
    retVal float[]; 
BEGIN 
    FOR I IN array_lower(arrFloats, 1)..array_upper(arrFloats, 1) LOOP 
    FOR K IN array_lower(x, 1)..array_upper(x, 1) LOOP 
     IF (arrFloats[I]= x[K])THEN 
     retVal[j] :=b; 
     j:=j+1; 
     retVal[j] :=c; 
     j:=j+1; 
     ELSE 
     retVal[j] := arrFloats[I]; 
     j:=j+1;  
     END IF; 
    END LOOP; 
    END LOOP; 
    RETURN retVal; 
END; 
$$ LANGUAGE plpgsql STABLE RETURNS NULL ON NULL INPUT; 

具体指标的值当我运行此查询替换的PostgreSQL 9.3

SELECT array_replace(array[1,20,2,5]); 

它给我这个样子

"[0:8]={1,1,20,20,2.22,3.33,2,5,5}" 

输出现在我不知道为什么要来此重复值。我的意思是马上嵌套循环...

我需要这样一个

"[0:8]={1,20,2.22,3.33,5}" 

回答

0

一个输出你必须与x阵列,具有两个元素的双回路。在每次迭代中,您都会将元素推送到结果数组中,因此您可以获得两倍的值。

如果我正确理解您的逻辑,您希望以相同的顺序扫描输入数组的值。如果相同,则将这些值替换为另一个数组,使其他值保持不变。有没有内置的功能,帮助你在这里,所以你必须从头做到这一点:

CREATE FUNCTION array_replace(arrFloats float[]) RETURNS float[] AS $$ 
DECLARE 
    searchArr float[] := ARRAY[1.,20.]; 
    replaceArr float[] := ARRAY[1.11,1.,111.,20.2,20.222]; 
    retVal  float[]; 
    i int; 
    ndx int; 
    len int; 
    upp int; 
    low int 
BEGIN 
    low := array_lower(searchArr, 1) 
    upp := array_upper(searchArr, 1); 
    len := upp - low + 1; 
    i := array_lower(arrFloats, 1); 
    WHILE i <= array_upper(arrFloats, 1) LOOP -- Use WHILE LOOP so can update i 
    ndx := i; -- index into arrFloats for inner loop 
    FOR j IN low .. upp LOOP 
     IF arrFloats[ndx] != searchArr[j] THEN 
     -- No match so put current element of arrFloats in the result and update i 
     retVal := retVal || arrFloats[i]; 
     i := i + 1; 
     EXIT; -- No need to look further, break out of inner loop 
     END IF; 
     ndx := ndx + 1; 
     IF j = upp THEN 
     -- We have a match so append the replaceArr to retVal and 
     -- increase i by length of search_array 
     retVal := retVal || replaceArr; 
     i := i + len; 
     END IF; 
    END LOOP; 
    END LOOP; 
    RETURN retVal; 
END; 
$$ LANGUAGE plpgsql STABLE STRICT; 

如果您做了searchArrreplaceArr成多个参数,以及该功能将变得更加灵活。

测试

[email protected]:~$ psql -d test 
psql (9.5.0, server 9.4.5) 
Type "help" for help. 

test=# select array_replace(array[1,20,2,5]); 
     array_replace 
------------------------------ 
{1.11,1,111,20.2,20.222,2,5} 
(1 row) 

test=# select array_replace(array[1,20,2,5,1,20.1,1,20]); 
         array_replace 
------------------------------------------------------------ 
{1.11,1,111,20.2,20.222,2,5,1,20.1,1.11,1,111,20.2,20.222} 
(1 row) 

正如你可以看到它的工作原理为搜索阵列的多个实例。

+0

首先感谢您的回应。关键是我需要一个嵌套循环,因为对于每个所需的数组,让我们说数组[1,20,2,5]);我会给喜欢maching值让说数组[1,20]和替换将是这样[1.11,1,111,20.2,20.222] ..所以这里的替换将像数组[1.11,1,111,20.2,20.222,2 ,5]);你认为没有嵌套循环可能吗? – zahid

+0

查看我的更新回答,输出 – Patrick

+0

感谢帕特里克为您的答案...亚很好.... – zahid