☰

첫 번째 열이 동일한 여러 열의 모든 행의 평균 찾기

tag-icon • tag-icon awk
첫 번째 열이 동일한 여러 열의 모든 행의 평균 찾기

포스팅의 연장선입니다첫 번째 열이 동일한 평균 행

입력 파일:

a   12  13  14
b   15  16  17
a   21  22  23
b   24  25  26

원하는 출력:

a   16.5  17.5  18.5
b   19.5  20.5  21.5

해당 게시물의 awk 코드는 다음과 같습니다.

awk '
    NR>1{
        arr[$1]   += $2
        count[$1] += 1
    }
    END{
        for (a in arr) {
            print a "\t" arr[a] / count[a]
        }
    }
'

문제: 이 코드는 첫 번째 행에서만 작동합니다. 이 코드를 여러 열로 확장하려면 어떻게 해야 합니까?

답변1

를 사용하면 awk키(첫 번째 열 값)와 열 인덱스에서 복합 인덱스를 구성하여 2D 배열을 시뮬레이션할 수 있습니다.

awk '
  {
  c[$1]++; 
  for (i=2;i<=NF;i++) {
    s[$1"."i]+=$i};
  } 
  END {
    for (k in c) {
      printf "%s\t", k; 
      for(i=2;i<NF;i++) printf "%.1f\t", s[k"."i]/c[k]; 
      printf "%.1f\n", s[k"."NF]/c[k];
    }
  }' file
  a       16.5    17.5    18.5
  b       19.5    20.5    21.5

perl배열의 해시를 사용하여 유사한 접근 방식을 보다 직접적으로 구현할 수 있습니다 .

또는 GNU datamash(적어도 버전 1.1.0부터)는 매우 컴팩트하게 그룹 평균화를 지원합니다.

datamash --sort --whitespace groupby 1 mean 2-4 < file
a       16.5    17.5    18.5
b       19.5    20.5    21.5

FWIW 여기에 의견에서 요청한 대로 전역 최대 평균으로 정규화하는 것을 포함하여 Perl 솔루션에 대한 나의 시도가 있습니다.면책 조항: 저는 초보자 Perl 프로그래머이므로 프로그래밍 방식이 좋지 않을 수 있습니다..

#!/usr/bin/perl

use strict;
use warnings;

use List::MoreUtils qw(pairwise minmax);
use Math::Round qw(nearest);

my @hdr;
my %sums = ();
my %count = ();
my $key;

while (defined($_ = <ARGV>)) {
  chomp $_;
  my @F = split(' ', $_, 0);

  # UGLY: hardcoded to expect exactly 1 header row
  if ($. == 1) {
    @hdr = @F;
    next;
  }

  # sum column-wise, grouped by first column
  $key = shift @F;
  if ( exists $sums{$key} ) {
    $sums{$key} = [ pairwise { $a + $b } @{ $sums{$key} }, @F];
  }
  else {
    $sums{$key} = \@F;
  }

  $count{$key}++;
}


my %avgs = ();
# NB should really initialize $maxavg to a suitably large NEGATIVE value
my $maxavg = 0.0;

# find the column averages, and the global max of those averages
for $key ( keys %sums ) {
  $avgs{$key} = [ map { $_ / $count{$key} } @{ $sums{$key} } ];
  # NB could use List::Util=max here, but we're alresdy using List::MoreUtils
  my ($kmin, $kmax) = minmax @{ $avgs{$key} };
  $maxavg = $kmax > $maxavg ? $kmax : $maxavg;
}

# normalize and print the results, rounded to nearest 0.01
print join "\t", @hdr, "\n";
for $key ( sort keys %avgs ) {
  print join "\t", $key, (map { nearest (0.01, $_ / $maxavg) } @{ $avgs{$key} }), "\n";
}

다른 이름으로 저장 colavgnorm.pl하고 실행한 다음 실행하세요.

$ ./colavgnorm.pl file
K       C1      C2      C3
a       0.77    0.81    0.86
b       0.91    0.95    1

file어디

K   C1  C2  C3
a   12  13  14
b   15  16  17
a   21  22  23
b   24  25  26

관련 정보