替换 postgresql 9.3 中特定索引的值

Question

CREATE OR REPLACE FUNCTION array_replace(INT[]) RETURNS float[] AS $$
DECLARE
  arrFloats ALIAS FOR ; 
  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 数组有两个元素。在每次迭代中，您将元素推入结果数组，因此您获得两倍的值。

如果我对你的逻辑理解正确，你想以相同的顺序扫描输入数组以获取另一个数组的值。如果相同，则将这些值替换为另一个数组，保留其他值不变。这里没有 built-in 函数可以帮助您，因此您必须从头开始：

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 中特定索引的值

replacing values of specific index in postgresql 9.3

postgresql

postgresql-9.3