在pl / r中将向量作为参数传递以启动for循环

时间:2018-12-12 00:39:44

标签: r for-loop vector arguments plr

需要使用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向量,它将起作用。

也尝试在函数内部声明为数组,文本,向量。

感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

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

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

这对我有用。多亏了我的老板:)