首页 > 解决方案 > 在 pl/r 中传递一个向量作为参数来启动一个 for 循环

问题描述

需要for loop使用. vector_ pl/r代码如下:

DROP TYPE IF EXISTS tsdata CASCADE;
CREATE TYPE tsdata
AS
(
  a text,
  b text,
  c text,
  d text 
);


DROP FUNCTION IF EXISTS standard.create_pd_ts(data char, dimnames char);
CREATE FUNCTION standard.create_pd_ts(data char, dimnames char)
  RETURNS setof tsdata AS
$$
  drv <- dbDriver("PostgreSQL")
  conn <- dbConnect(drv, dbname = 'abc')

  iq <- paste0("select * from data")
  data <-dbGetQuery(conn, iq)
  data <- data.frame(coef_db_res_tab)

  for(i in 1:length(dimnames)) {
      var = dimnames [i]
      uniquedim= unique(data[,i])
      assign(paste0('dim',i),uniquedim)
     }
  ## save dimensions in a vector
  dimvec<-paste0('dim',1:length(dimnames),collapse = ',')

  tsdata <- eval(parse(text=paste0("expand.grid(", dimvec, 
  ",stringsAsFactors = FALSE)")))
  return(tsdata)

$$
LANGUAGE 'plr';

select
  (q).a,    
  (q).b,
  (q).c,
  (q).d
from
  (
    select * from standard.create_pd_ts(‘data’, 'c("a","b","c","d")')
  ) q ;

该函数的动机是找到 a、b、c 和 d 列的唯一值;此后,将相同的值存储在 dimvec 向量中,然后使用每列的唯一值来扩展网格以创建所有组合。尝试了各种方法,但 for 循环没有启动。如果我在函数内声明 dimnames 向量,它就可以工作。

还尝试在函数内声明为数组、文本、声明为向量。

任何帮助表示赞赏。

标签: rfor-loopvectorargumentsplr

解决方案


select
  (q).a,    
  (q).b,
  (q).c,
  (q).d
from
  (

select * from standard.create_pd_ts('data', ARRAY['a','b','c','d'])  
  ) q ;

这对我有用。感谢我的老板:)


推荐阅读