Маніпулювати деякими погано обмеженими даними в корисному CSV


13

У мене є деякий вихід у вигляді:

count  id     type
588    10 |    3
 10    12 |    3
883    14 |    3
 98    17 |    3
 17    18 |    1
77598    18 |    3
10000    21 |    3
17892     2 |    3
20000    23 |    3
 63    27 |    3
  6     3 |    3
 2446    35 |    3
 14    4 |    3
 15     4 |    1
253     4 |    2
19857     4 |    3
 1000     5 |    3
...

Що досить безладно, і його потрібно прибрати до CSV, щоб я міг подарувати його керівнику проектів для них, електронну таблицю, на біса.

Суть проблеми полягає в цьому: мені потрібно, щоб результат цього був:

id, sum_of_type_1, sum_of_type_2, sum_of_type_3

Прикладом цього є id "4":

14    4 |    3
 15     4 |    1
253     4 |    2
19857     4 |    3

Натомість це повинно бути:

4,15,253,19871

На жаль, я дуже сміття в цьому роді, мені вдалося прибрати всі рядки до очищення та в CSV, але мені не вдалося подвоїти і згрупувати рядки. Зараз у мене це є:

awk 'BEGIN{OFS=",";} {split($line, part, " "); print part[1],part[2],part[4]}' | awk '{ gsub (" ", "", $0); print}'

Але все, що потрібно зробити - це очистити сміття з символів і знову надрукувати рядки.

Який найкращий спосіб отримати масаж рядків у згаданому вище висновку?


Ви навіть хочете підсумовувати підрахунки разом?
hjk

Відповіді:


12

Спосіб зробити це - скласти все в хеш.

# put values into a hash based on the id and tag
awk 'NR>1{n[$2","$4]+=$1}
END{
    # merge the same ids on the one line
    for(i in n){
        id=i;
        sub(/,.*/,"",id);
        a[id]=a[id]","n[i];
    }
    # print everyhing
    for(i in a){
        print i""a[i];
    }
}'

редагувати: моя перша відповідь не відповіла на питання належним чином


Так, це дуже добре зробило трюк. Дякую! Єдине, що я не врахував, що деякі типи ідентифікаторів будуть порожніми і, таким чином, псують CSV, але я можу опрацювати цю дрібницю
Пол,

@Paul Можливо, додамо NF<4{$4="no_type";}на старті
DarkHeart

11

Перл на допомогу:

#!/usr/bin/perl
use warnings;
use strict;
use feature qw{ say };

<>;  # Skip the header.

my %sum;
my %types;
while (<>) {
    my ($count, $id, $type) = grep length, split '[\s|]+';
    $sum{$id}{$type} += $count;
    $types{$type} = 1;
}

say join ',', 'id', sort keys %types;
for my $id (sort { $a <=> $b } keys %sum) {
    say join ',', $id, map $_ // q(), @{ $sum{$id} }{ sort keys %types };
}

Він зберігає дві таблиці, таблицю типів та таблицю ідентифікаторів. Для кожного id він зберігає суму за типом.


5

Якщо GNU datamash - це варіант для вас, то

awk 'NR>1 {print $1, $2, $4}' OFS=, file | datamash -t, -s --filler=0 crosstab 2,3 sum 1
,1,2,3
10,0,0,588
12,0,0,10
14,0,0,883
17,0,0,98
18,17,0,77598
2,0,0,17892
21,0,0,10000
23,0,0,20000
27,0,0,63
3,0,0,6
35,0,0,2446
4,15,253,19871
5,0,0,1000

4

Python (і pandasбібліотека зокрема дуже підходить для такої роботи

data = """count  id     type
588    10 |    3
 10    12 |    3
883    14 |    3
 98    17 |    3
 17    18 |    1
77598    18 |    3
10000    21 |    3
17892     2 |    3
20000    23 |    3
 63    27 |    3
  6     3 |    3
 2446    35 |    3
 14    4 |    3
 15     4 |    1
253     4 |    2
19857     4 |    3
 1000     5 |    3"""

import pandas as pd
from io import StringIO # to read from string, not needed to read from file

df = pd.read_csv(StringIO(data), sep=sep='\s+\|?\s*', index_col=None, engine='python')

Це зчитує дані csv до a pandas DataFrame

    count  id  type
0     588  10     3
1      10  12     3
2     883  14     3
3      98  17     3
4      17  18     1
5   77598  18     3
6   10000  21     3
7   17892   2     3
8   20000  23     3
9      63  27     3
10      6   3     3
11   2446  35     3
12     14   4     3
13     15   4     1
14    253   4     2
15  19857   4     3
16   1000   5     3

Потім групуємо ці дані по idі беремо суму стовпцяcount

df_sum = df.groupby(('type', 'id'))['count'].sum().unstack('type').fillna(0)

unstack Перебудовує це , щоб перемістити ідентифікатори до колон, і fillnaзаповнює порожні поля з 0 '

df_sum.to_csv()

Це повертається

id,1,2,3
2,0.0,0.0,17892.0
3,0.0,0.0,6.0
4,15.0,253.0,19871.0
5,0.0,0.0,1000.0
10,0.0,0.0,588.0
12,0.0,0.0,10.0
14,0.0,0.0,883.0
17,0.0,0.0,98.0
18,17.0,0.0,77598.0
21,0.0,0.0,10000.0
23,0.0,0.0,20000.0
27,0.0,0.0,63.0
35,0.0,0.0,2446.0

Оскільки кадр даних містить відсутні дані (порожні комбінації типів id), панди перетворюють ints в float(обмеження внутрішніх опрацювань) Якщо ви знаєте, що входи будуть лише int, ви можете змінити наступний рядок наdf_sum = df.groupby(('type', 'id'))['count'].sum().unstack('type').fillna(0).astype(int)


1
Ви повинні пояснити, що робить код, який ви надали, тому він корисний усім, хто бачить цю публікацію, а не цій конкретній особі.
Фонд позову Моніки

Це зрозуміліше? Я також виправив
регекс для сеператора

Мені добре виглядає. Дякуємо, що додали пояснення!
Позов по

3

Ви можете використовувати Perl для циклічного перегляду CSV-файлу та накопичення суми відповідних типів у хеші, перебуваючи в дорозі. І врешті-решт виведіть інформацію, зібрану для кожного посвідчення особи.

Структура даних

%h = (
   ID1    =>  [ sum_of_type1, sum_of_type2, sum_of_type3 ],
   ...
)

Це допомагає зрозуміти код нижче:

Perl

perl -wMstrict -Mvars='*h' -F'\s+|\|' -lane '
   $, = chr 44, next if $. == 1;

   my($count, $id, $type) = grep /./, @F;
   $h{ $id }[ $type-1 ] += $count}{
   print $_, map { $_ || 0 } @{ $h{$_} } for sort { $a <=> $b } keys %h
' yourcsvfile

Вихідні дані

2,0,0,17892
3,0,0,6
4,15,253,19871
5,0,0,1000
...

1

моє взяти, не надто відрізняється від інших. Використовує GNU awk, який має масиви масивів

gawk '
    NR == 1 {next}
    {count[$2][$4] += $1}
    END {
        for (id in count) {
            printf "%d", id
            for (type=1; type<=3; type++) {
                # add zero to coerce possible empty string into a number 
                printf ",%d", 0 + count[id][type]
            }
            print ""        # adds the newline for this line
        }
    }
' file

виходи

2,0,0,17892
3,0,0,6
4,15,253,19871
5,0,0,1000
10,0,0,588
12,0,0,10
14,0,0,883
17,0,0,98
18,17,0,77598
21,0,0,10000
23,0,0,20000
27,0,0,63
35,0,0,2446

0

Ви можете використовувати цей код для підсумовування значень на основі стовпця id,

Я додав один код awk після вашого коду

awk 'BEGIN{OFS=",";} {split($line, part, " "); print part[1],part[2],part[4]}' abcd | awk '{ gsub (" ", "", $0); print}' | awk 'BEGIN{FS=OFS=SUBSEP=","}{arr[$2,$3]+=$1;}END{for ( i in arr ) print i,arr[i];}'

Продовжуйте це ...

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.