如何将不同文件中的一列与另一列求和?

如何将不同文件中的一列与另一列求和?

我有这样的东西

文件1:

element1,12
element2,10
element3,5

文件2:

IN
element1,8
element3,5

我想修改file1并获取:

element1,20
element2,10
element3,10

file2 的第一行包含“IN”,这意味着我想进行添加。如果它是“OUT”,则应减去这些值,并且 file1 应如下所示:

element1,4
element2,10
element3,0

我必须使用 awk、grep、sed。多谢!

答案1

$ cat tst.awk
BEGIN { FS=OFS="," }
NR==FNR {
    if ( FNR == 1 ) {
        mult = ($1=="IN" ? 1 : -1)
    }
    else {
        add[$1] = mult * $2
    }
    next
}
{
    $2 += add[$1]
    print
}

$ awk -f tst.awk file2 file1
element1,20
element2,10
element3,10

$ awk -f tst.awk file3 file1
element1,4
element2,10
element3,0

$ head file{1..3}
==> file1 <==
element1,12
element2,10
element3,5

==> file2 <==
IN
element1,8
element3,5

==> file3 <==
OUT
element1,8
element3,5

相关内容