主页 / dba / 问题 / 195448
Accepted
Noxville
Asked: 2018-01-16 17:52:14 +0800 CST

将相同大小的postgresql数组的元素相乘

  • 772

在 Postgresql 9.6 中，是否有一种干净的方法可以将两个相同大小的数组中的对应元素相乘？

例如

a = [1,2,9,4,5]
b = [5,8,4,0,12]

function(a, b) would return [5, 16, 36, 0, 60].

我更喜欢它作为这样的功能，因为它会在多个地方使用。

postgresql array

1 个回答

  1. Best Answer

    并行不嵌套

    你想要的是类似这样的东西parallel-unnest。这里我们使用ARRAY()构造函数而不是array_agg()

    SELECT ARRAY(
  SELECT a*b
  FROM unnest(
    ARRAY[1,2], -- ex1
    ARRAY[2,3]  -- ex2
  ) AS t(a,b)
);

    创建我们自己的函数

    在函数式编程中，我们称之为 this zip，您可以在具有原始功能支持的语言（如 Python 等）中使用它。在具有更高级功能支持的语言中，例如Haskell，您可以zipWith使用Ramda.jsR.zipWith()

    ex1这将返回一个大小为或中较大者的数组ex2。你可以把它放在这样的函数中，

    CREATE FUNCTION array_zip_with_multiply( arr_q anyarray, arr_e anyarray )
RETURNS anyarray
AS $$
  SELECT ARRAY(
    SELECT a*b
    FROM unnest(
      arr_q, -- ex1
      arr_e  -- ex2
    ) AS t(a,b)
  );
$$ LANGUAGE sql
IMMUTABLE;

SELECT array_zip_with_multiply( ARRAY[1,2,9,4,5], ARRAY[5,8,4,0,12] );
 array_zip_with_multiply 
-------------------------
 {5,16,36,0,60}
(1 row)

    *注意：如果没有为该类型定义运算符，这将不起作用。

    • 4

相关问题