用满足其他列条件的值填充列

Question

我有如下表格：

con_no     pr_no     flag_1  flag_2
con_002   1000234      1       0
con_002   1000345     -1       0
con_002   1100200      1       1
con_002   1005890      0       0
con_003   1100367      0       0
con_003   1000445      1       1
con_003   1200334     -1       0
con_003   1140567      1       0

现在我想要另一列填充 pr_no，其 flag_2 = 1 并且 flag_1 = 1 在特定 con_no 内。所以结果表应该是这样的

con_no     pr_no     flag_1  flag_2  Parent_pr
con_002   1000234      1       0      1100200 <---This belongs to con_002
con_002   1000345     -1       0      
con_002   1100200      1       1      1100200 <---This belongs to con_002
con_002   1005890      0       0
con_003   1100367      0       0
con_003   1000445      1       1      1000445 <---This belongs to con_003
con_003   1200334     -1       0
con_003   1140567      1       0      1000445 <---This belongs to con_003

如何使用 python (pandas) 或 SQL 实现上述功能。实际上这个表在数据库（postgreSQL）中。所以一个 SQL 查询字符串也可以。

但有一个问题。每次程序运行时，con_no 都会重新洗牌。一些pr_no 可能附加到新的或现有的（但不同的con_no）。

Answer 1

您应该将表与自身连接起来。如果您的数据位于名为my_table 的表中，则应该可以使用这样的查询：

with cte as (
    select con_no, pr_no
       from my_table
      where flag_2 = 1)
select t.con_no, t.pr_no, t.flog_1, t.flog_2, cte.pr_no as parent_pf
  from my_table as t
  left outer join cte on (cte.con_no = t.con_no and t.flag_1 = 1)

Answer 2

您可以使用窗口函数和条件逻辑：

select t.*,
       (case when flag_1 = 1
             then max(case when flag_1 = 1 and flag_2 = 1 then pr_no end) over (partition by con_no)
        end) as parent_pr
from t;

实际上，Postgres 支持filter 语法，所以我会这样写：

select t.*,
       (case when flag_1 = 1
             then max(pr_no) filter (where flag_1 = 1 and flag_2 = 1) over (partition by con_no)
        end) as parent_pr
from t;