Question

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;

当我运行此查询时

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}"

Answer 1

你有一个双循环，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;

如果您将searchArr和replaceArr纳入参数，此功能也会变得更加灵活。

<强>测试

patrick@puny:~$ 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)

正如您所看到的，它适用于多次出现的搜索数组。

替换postgresql 9.3中特定索引的值

1 个答案: