xiaodai's questions

我有一列A，我想根据列B和来增加或减少C。

逻辑是这样的：

• 如果lag(A) = 0且如果B= 1 则A=1
• 如果lag(A) = 1且如果C= 1 则A=0
• 来自lag(A)ordering byD

我不知道如何在 DuckDB 中实现此逻辑。我认为问题在于这A取决于前一个值。

select
  unnest(array_agg(row(B, C) order by D)) as list1
from
  my_table

我认为非嵌套版本一定有某种用途list1，我认为list_reduce可能有用，但我搞不清楚如何解决。另一个想法是递归 CTE 可能有用。但我找不到解决方案。

也许在 DuckDB 中这是不可能的，我需要使用列表 UDF。

如果在 SQL 的任何其他变体中也可行，那么我也很想听到答案。

示例数据

   B C A  D
1  0 0 0  1
2  1 0 1  2
3  0 0 1  3
4  1 0 1  4
5  0 1 0  5
6  1 0 1  6
7  1 0 1  7
8  0 1 0  8
9  0 0 0  9
10 0 1 0 10

R 代码生成数据

fn = function(b, c) {
  l = length(b)
  A = vector(mode="integer", l)
  A[1] = 0
  for (i in 2:l) {
    A[i]= A[i-1]
    if((b[i] == 1) & (A[i-1] == 0)) {
      A[i] = 1
    } else if ((c[i] == 1) & (A[i-1] == 1)) {
      A[i] = 0
    }
  }
  A
}


d = data.frame(
  B = c(0, 1, 0, 1, 0, 1, 1, 0, 0, 0),
  C = c(0, 0, 0, 0, 1, 0, 0, 1, 0, 1)
) %>% dplyr::mutate(
  A = fn(B, C)
) %>% 
  mutate(D = row_number())

d

考虑以下代码：

library(dplyr)
a = data.frame(a = 1:3)
abc = \(x) data.frame(x/2, x/3)
a %>% 
  mutate(abc(a))

返回

  a x.2       x.3
1 1 0.5 0.3333333
2 2 1.0 0.6666667
3 3 1.5 1.0000000

但我无法控制名称，因为我想为它们分配其他名称。

解决这个问题的一种方法是将返回类型abc命名为，但这并不理想，因为我可能想要abc在两组变量上运行

a %>% 
  mutate(abc(a), abc(b))

另一种方法当然是rename在每次运行后都进行，abc但这可能会不稳定。

有没有办法给输出命名abc？例如

a %>% 
  mutate(c("a1","a2") := abc(a), c("b1","b2") := abc(b))

但上述:=语法不起作用。

我今天必须做这样的事情

data1 %>%
  summarise(
   ab1 = fn(a1, b1),
   ab2 = fn(a2, b2), 
   ab3 = fn(a3, b3) 
  )
# imagine if there are 100 of them

如果fn是单参数函数我可以这样做

data1 %>%
  summarise(across(starts_with("a", fn)))

但不幸的是，我的函数需要两列作为输入。有没有更优雅的方法来实现这一点？