☰

基於公共列合併多個文件

tag-icon • tag-icon shell awk merge join
基於公共列合併多個文件

我有多個文件，每個文件都有兩列：

例如 ：

文件1.txt

ID Value1
1   40
2   30
3   70

文件2.txt

ID Value2
3   50
4   70 
9   20

等等，

文件1230.txt

ID Value150
9   98
10  52
2   71

如何根據第一列（基於文件之間的交集）合併這些文件？我的輸出應該是

ID Value1 Value2 Value150
1   40     0      0
2   30     0      71
3   70     50     0
4   0      70     0
9   0      20     98
10  0      0      52

有人可以使用 awk 或 Linux 指令來幫助解決這個問題嗎？

謝謝。

答案1

這是使用 python 實現此目的的一種方法。

代碼：

import sys

columns = []
data = {}
ids = set()
for filename in sys.argv[1:]:
    with open(filename, 'rU') as f:
        key = next(f).strip().split()[1]
        columns.append(key)
        data[key] = {}
        for line in f:
            if line.strip():
                id, value = line.strip().split()
                try:
                    data[key][int(id)] = value
                except ValueError as exc:
                    raise ValueError(
                        "Problem in line: '{}' '{}' '{}'".format(
                            id, value, line.rstrip()))

                ids.add(int(id))

print('\t'.join(['ID'] + columns))

for id in sorted(ids):
    line = []
    for column in columns:
        line.append(data[column].get(id, '0'))
    print('\t'.join([str(id)] + line))

結果：

ID  Value1  Value2  Value150
1   40  0   0
2   30  0   71
3   70  50  0
4   0   70  0
9   0   20  98
10  0   0   52

答案2

帶有命令列工具的 Bash 解決方案。輸入檔列表無序，因此ls -v輸出到cat.

while read line; do
    if [[ "$line" =~ ID ]]; then
        array=${line##* }
        index+=($array)
        continue
    else
        eval $array'[${line% *}]=${line#* }'
    fi
done <<<"$( cat $(ls -v file[0-9]*.txt) )"

printf ID
for name in ${index[@]}; do
    printf ' %s' $name
done
echo

max_ind=$( sort -nu file[0-9]*.txt | tail -n1 | cut -d' ' -f1 )

for (( j = 1 ; j <= $max_ind ; j++ )); do
    for (( i = 0 ; i < ${#index[@]} ; i++ )); do
        value=$( eval 'echo ${'${index[i]}'[j]}' )
        roll+=$( [ "$value" ] &&
            printf "%-${#index[i]}s " $value ||
            printf "%-${#index[i]}s " 0 )
    done
    [[ "$roll" =~ [^0\ ] ]] && printf '%-4s%s\n' $j "$roll"
    unset roll
done

相關內容