awk根据其他列的条件创建新变量

Question

我有一个具有以下结构的文件input.txt （〜100'000行）：

Z0        Z1        Z2
0.9746    0.0254    0.0000     
0.0032    0.0000    0.9433  
0.2464    0.5603    0.9008 
0.4034    0.4982    0.0069 
0.0072    0.9996    0.0472 
...       ...       ...

我想根据以下条件创建一个名为SCORE的附加列的新文件output.txt ：

• 如果：0.17≤Z0≤0.33和0.40≤Z1≤0.60，则SCORE = 1
• 如果：0.40≤Z0≤0.60和0.40≤Z1≤0.60，则SCORE = 2
• 如果Z0≤0.05且Z1≥0.95且Z2≤0.05则SCORE = 3
• 如果Z0≤0.05且Z1≤0.05且Z2≥0.95，则SCORE = 4
• 如果其他4个条件不适用，则SCORE = 5。

output.txt看起来像这样：

Z0        Z1        Z3         SCORE
0.9746    0.0254    0.0000     5
0.0032    0.0000    0.9433     5
0.2464    0.5603    0.9008     1
0.4034    0.4982    0.0069     2
0.0072    0.9996    0.0472     3           
...       ...       ...

这是我尝试过的：

awk 'NR==1{$4="SCORE";print;next} \
  0.17<=$1 && $1<=0.33 && 0.40<=$2 && $2<=0.60 {$4="1"} \
  0.40<=$1 && $1<=0.60 && 0.40<=$2 && $2<=0.60 {$4="2"} \
  $1<=0.05 && $2>=0.95 && $3<=0.05 {$4="3"} \
  $1<=0.05 && $2<=0.05 && $3>=0.95 {$4="4"} \
  *other* 1' input.txt > output.txt

但是，在前5个命令行中出了点问题，我不知道如何在最后一行中写入最后一个条件（对于得分5）。

Answer 1

这样的事？

NR==1{$4="SCORE";print;next}
0.17<$1  && $1<0.33  && 0.40<$2  && $2<0.60 {print $0, "1";next}
0.40<$1  && $1<0.60  && 0.40<$2  && $2<0.60 {print $0, "2";next}
$1<=0.05 && $2>=0.95 && $3<=0.05            {print $0, "3";next}
$1<=0.05 && $2<=0.05 && $3>=0.95            {print $0, "4";next}
{print $0, "5"}

（输出的第2行中可能存在错误，因为0.9433小于0.95）

Answer 2

不需要反斜杠和引用数字...您所描述的条件和代码不匹配（ <= vs < ）。

$ awk 'NR==1{print $0,"SCORE"; next} 
  {score=5}
  0.17<$1 && $1<0.33 && 0.40<$2 && $2<0.60 {score=1}
  0.40<$1 && $1<0.60 && 0.40<$2 && $2<0.60 {score=2}
  $1<=0.05 && $2>=0.95 && $3<=0.05         {score=3}
  $1<=0.05 && $2<=0.05 && $3>=0.95         {score=4}  
  {print $0,score}' file | column -t

Z0      Z1      Z3      SCORE
0.9746  0.0254  0.0000  5
0.0032  0.0000  0.9433  5
0.2464  0.5603  0.9008  1
0.4034  0.4982  0.0069  2
0.0072  0.9996  0.0472  3

您的第二行预期得分也是错误的。

或者，也许合并相同的条件

$ awk 'NR==1{print $0,"SCORE"; next} 
  {score=5}
  0.40<$2 && $2<0.60 {if(0.17<$1 && $1<0.33)   score=1;
                      if(0.40<$1 && $1<0.60)   score=2}
  $1<=0.05           {if($2>=0.95 && $3<=0.05) score=3;
                      if($2<=0.05 && $3>=0.95) score=4}  
  {print $0,score}' file | column -t

Answer 3

我认为您显示的输出不正确，请检查一次，能否请您尝试关注一下，并告诉我这是否对您有帮助。

awk '
function check(line,Z0,Z1,Z3){
  if(0.17 <= Z0 <= 0.33 && 0.40 <= Z1 <= 0.60)           {   score=1  }
  else if(0.40 <= Z0 <= 0.60 && 0.40 <= Z1 <= 0.60)      {   score=2  }
  else if(Z0 <= 0.05 && Z1 >= 0.95 && Z3 <= 0.05)        {   score=3  }
  else if(Z0 <= 0.05 && Z1 <= 0.05 && Z3 >= 0.95)        {   score=4  }
  else                                                   {   score=5  }
  return score}
FNR==1{
  print;
  next}
{
  check($0,$1,$2,$3);
  print $0,score
}
' input.txt | column -t