我有以下帶有一列sig和N其他列的資料框。
sig包含N嵌入其中的列數，如下所示。嵌入的列名稱可以是資料框中存在的任何數字。
我想sig用其他列中的相應值更新該列。
例如，
--------------------------------------------------------------------- ------------ ------------------ ------------------- --------
|sig |order_timing|po_manl_create_ind|mabd_arrival_status|cut_time|
--------------------------------------------------------------------- ------------ ------------------ ------------------- --------
|R1:BR1-order_timing:BR2-po_manl_create_ind:BR3-mabd_arrival_status:R1|14 |0 |late |23 |
|R1:BR1-order_timing:BR2-po_manl_create_ind:BR7-cut_time:R1 |14 |0 |on_time |10 |
預期輸出
--------------------------------------------------------------------- ------------ -----
|sig |order_timing|po_manl_create_ind|mabd_arrival_status|cut_time|
--------------------------------------------------------------------- ------------ -----
|R1:BR1-14:BR2-0:BR3-late:R1|14 |0 |late |23 |
|R1:BR1-14:BR2-0:BR7-10:R1 |14 |0 |on_time |10 |
uj5u.com熱心網友回復：
一種方法是replace使用可能出現在sig值中的列串列來鏈接多個運算式。
使用此示例 DF：
val df = Seq(
("R1:BR1-order_timing:BR2-po_manl_create_ind:BR3-mabd_arrival_status:R1", 14, 0, "late", 23),
("R1:BR1-order_timing:BR2-po_manl_create_ind:BR7-cut_time:R1", 14, 0, "on_time", 10),
).toDF("sig", "order_timing", "po_manl_create_ind", "mabd_arrival_status", "cut_time")
您可以replace_expr使用 foldLeft生成替換運算式，如下所示：
val replace_expr = df.columns
.filter(_ != "sig")
.foldLeft("sig")((acc, c) => s"replace($acc, '$c', $c)")
df.withColumn("sig", expr(replace_expr)).show(false)
// --------------------------- ------------ ------------------ ------------------- --------
//|sig |order_timing|po_manl_create_ind|mabd_arrival_status|cut_time|
// --------------------------- ------------ ------------------ ------------------- --------
//|R1:BR1-14:BR2-0:BR3-late:R1|14 |0 |late |23 |
//|R1:BR1-14:BR2-0:BR7-10:R1 |14 |0 |on_time |10 |
// --------------------------- ------------ ------------------ ------------------- --------
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/361515.html