Обработка двух файлов с использованием awk

9

Я прочитал Сравнение двух файлов с использованием Unix и Awk . Это действительно интересно. Я прочитал и проверил это, но я не могу понять это полностью и использую это в других случаях.

У меня есть два файла. file1имеет одно поле, а другое имеет 16 полей. Я хочу прочитать элементы file1 и сравнить их с 3-м полем file2. Если для каждого элемента было совпадение, я суммирую значение поля 5 в file2. Например:

файл 1

1
2
3

файл 2

2 2 2 1 2
3 6 1 2 4 
4 1 1 2 3
6 3 3 3 4 

Для элемента 1 в file1я хочу добавить значения в поле 5, file2где значение поля 3 равно 1. И сделать то же самое для элементов 2 и 3 в file1. Выход для 1 равен (3 + 4 = 7), для 2 - 2, а для 3 - 4.

Я не знаю, как мне написать это с помощью awk.

user55340
источник

Ответы:

20

Вот один из способов. Я написал его как скрипт awk, чтобы я мог добавлять комментарии:

#!/usr/local/bin/awk -f

{
    ## FNR is the line number of the current file, NR is the number of 
    ## lines that have been processed. If you only give one file to
    ## awk, FNR will always equal NR. If you give more than one file,
    ## FNR will go back to 1 when the next file is reached but NR
    ## will continue incrementing. Therefore, NR == FNR only while
    ## the first file is being processed.
    if(NR == FNR){
      ## If this is the first file, save the values of $1
      ## in the array n.
      n[$1] = 0
    }
    ## If we have moved on to the 2nd file
    else{
      ## If the 3rd field of the second file exists in
      ## the first file.
      if($3 in n){
        ## Add the value of the 5th field to the corresponding value
        ## of the n array.
        n[$3]+=$5
      }
    }
}
## The END{} block is executed after all files have been processed.
## This is useful since you may have more than one line whose 3rd
## field was specified in the first file so you don't want to print
## as you process the files.
END{
    ## For each element in the n array
    for (i in n){
    ## print the element itself and then its value
    print i,":",n[i];
    }
}

Вы можете сохранить это как файл, сделать его исполняемым и запустить его так:

$ chmod a+x foo.awk
$ ./foo.awk file1 file2
1 : 7
2 : 2
3 : 4

Или вы можете сжать его в одну строку:

awk '
     (NR == FNR){n[$1] = 0; next}
     {if($3 in n){n[$3]+=$5}}
     END{for (i in n){print i,":",n[i]} }' file1 file2
Тердон
источник
9
awk '
  NR == FNR {n[$3] += $5; next}
  {print $1 ": " n[$1]}' file2 file1
Стефан Шазелас
источник
Это делает некоторую дополнительную работу, суммируя несопоставленные поля.
Эммануэль
@ Emmanuel, это по-прежнему одна инструкция awk на строку файла2, что делает его короче и быстрее, чем у тердона
Стефан Шазелас
блестящий раствор!
Рональд Пауфферт