ОСУЩЕСТВЛЕНИЮ `sumproduct` в UNIX оболочки

Question

У меня есть выход из сценария thescript, который гласит:

202 1 0 1 0 0 0 

Теперь я хочу, чтобы выборочно просуммировать число с awk, в зависимости от стоимости ${SUM_MASK}:

SUM_MASK=1,1,0,0,0,0,0 

Я хотел бы иметь что-то вроде:

thescript | awk <SOMETHING> 

где выход каждое число thescript умножается на соответствующее число в ${SUM_MASK}, получение:

203 

как результат: 203 = 202 * 1 + 1 * 1 + 0 * 0 + 1 * 0 + 0 * 0 + 0 * 0 + 0 * 0

Это было бы похоже на функцию sumproduct в программном обеспечении для работы с электронными таблицами.

Следующие snipets код сделать трюк, но я хотел бы избежать использования подстановки процессов:

SUM_MASK="1,1,0,0,0,0,0"; paste <(thescript) <(echo ${SUM_MASK} | tr ',' '\n') | awk '{ SUM += $1 * $2 } END { print SUM }' 

и именованные каналы:

SUM_MASK="1,1,0,0,0,0,0"; mkfifo fA; mkfifo fB; thescript > fA & echo ${SUM_MASK} | tr ',' '\n' > fB & paste fA fB | awk '{ SUM += $1 * $2 } END { print SUM }' > result.text; rm -f fA fB 

, как я мог бы достичь этого?

Answer 1

echo "202 1 0 1 0 0 0" | 
awk -v summask="1,1,0,0,0,0,0" ' 
    BEGIN {split(summask, mask, /,/)} 
    { sumproduct=0 
     for (i=1; i<=NF; i++) { 
      sumproduct += $i * mask[i] 
     } 
     print sumproduct 
    } 
' 

203 

Answer 2

Здесь нет необходимости в внешних инструментах, таких как awk - bash способен разрешать это только с помощью встроенных возможностей. Рассмотрим ниже реализацию в виде функции:

sumproduct() { 
    local -a sum_inputs sum_mask 
    local idx result 

    # read your sum_inputs into an array from stdin 
    IFS=', ' read -r -a sum_inputs # this could be <<<"$1" to use the first argument 

    # and your sum_mask from the like-named variable 
    IFS=', ' read -r -a sum_mask <<<"$SUM_MASK" # or <<<"$2" for the second argument 

    # ...iterate over array elements in sum_inputs; find the corresponding sum_mask; math. 
    result=0 
    for idx in "${!sum_inputs[@]}"; do 
    ((result += ${sum_mask[$idx]} * ${sum_inputs[$idx]})) 
    done 
    echo "$result" 
} 

---

Чтобы проверить это:

echo "202 1 0 1 0 0 0" | SUM_MASK=1,1,0,0,0,0,0 sumproduct 

... правильно дает:

203 

Answer 3

Вам действительно не нужен сумма продукта, но суммирование маскируется, например, это должно быть быстрее, если у вас много скрытых столбцов.

$ awk -v mask='1,1,0,0,0,0,0' 'BEGIN {n=split(mask,m,","); 
             for(i=1; i<=n; i++) if(m[i]) ix[i]} 
            {sum=0; 
             for(i in ix) sum += $i; 
             print sum}' file 

203 

Answer 4

С однозначных множителей можно сделать простой цикл

SUM_MASK=1,1,0,0,0,0,0 
offset=0 
sum=0; 
for i in 202 1 0 1 0 0 0; do 
    j="${SUM_MASK:$offset:1}" 
    ((sum += i * j)) 
    ((offset+=2)) 
done 
echo "${sum}" 

Этот раствор может быть использован в сценарии prodsum, который можно назвать с thescript | prodsum:

offset=0 
sum=0; 
for i ; do 
    j="${SUM_MASK:$offset:1}" 
    ((sum += i * j)) 
    ((offset+=2)) 
done 
echo "${sum}" 

EDIT : Когда SUB_MASK может иметь номера> 9, используйте следующее:

SUM_MASK=1,10,0,0,0,0,0 
sum=0; 
remaining_mask="$SUM_MASK" 

for i in 202 1 0 1 0 0 0; do 
    j="${remaining_mask%%,*}" 
    remaining_mask="${remaining_mask#*,}" 
    ((sum += i * j)) 
done 
echo "${sum}"