美文网首页
awk实现类sql的join操作

awk实现类sql的join操作

作者: xindoo | 来源:发表于2020-11-07 17:15 被阅读0次

    awk、grep和sed被称为linux三剑客,事实上grep和awk我在日常工作中也常用到(sed用的比较少),可能有些人对awk了解比较少,我先大概介绍下。 很多人以为awk只是一个文本处理工具,实际上他们也是这么用的。但事实上它其实是一门语言,拥有数学运算符、进程控制语句,甚至针对于文本处理封装了很多内置变量和函数,这造就了它强大的文本处理能力。 如果grep只能做到数据的筛选,那awk还能做到数据的处理、分析甚至生成报表,毕竟它是一门完整的编程语言。

    因为这篇文章不是awk的入门教程,如果想入门的话我推荐阮一峰老师的awk 入门教程 和 左耳朵耗子的AWK 简明教程

    回到我今天的正题,今天我给大家看个我常使用awk的场景。因为做后端开发,经常在做数据分析的时候会有这样的问题,1.面对几十万条的数据,需要筛选出几百几千个特定key的数据。2.对这几百万条数据,针对其中的id字段补齐其他字段。

    这时候可能有精通excel的同学跳出来说 “就这,so easy,vlookup就搞定!” ,事实上,excel确实可以解决问题,但有点重,甚至有些时候我们在服务器上还用不了excel。还有啥其他方法?针对我说的这俩场景,其实仔细想一下,是不是sql中俩表join就能解决问题。实际上你并不需要真正把文件灌到数据库里,只需要用awk一条命令就能解决。

    实例

    我们把问题具像下,假设有两个文件,score.txt存着学号+成绩的数据,另外一个name.txt存着学号+姓名的数据,你现在想知道每个人都烤了多少分。

    score.txt

    id score 
    1 87
    2 67
    3 68
    4 75
    5 90
    6 100
    7 0
    

    name.txt

    id name 
    1 张三
    2 李四
    3 王五
    4 赵二
    5 刘能
    6 熊大
    

    你想得到一份包含学号 姓名和成绩的数据,就像下面这样。

    id score name 
    1 87 张三
    2 67 李四
    3 68 王五
    4 75 赵二
    5 90 刘能
    6 100 熊大
    7 0
    

    用awk生成这样的数据有多简单?只需要一行代码,你可以保存name.txt和score.txt,然后执行下面命令尝试下。

    awk 'ARGV[1]==FILENAME {map[$1]=$2} ARGV[2]==FILENAME {print $0, map[$1]}' name.txt score.txt  
    

    简简单单就实现了name.txt和score.txt在id之上的right join。

    解释下上面的代码,ARGV和FILENAME是awk内置的变量,ARGV里存放在awk所接受的参数列表,像上面 ARGV[1]就是"name.txt",ARGV[2]就是"score.txt"。awk是面向行的,所以针对每一行数据都会执行ARGV[1]==FILENAME {map[$1]=$2} ARGV[2]==FILENAME {print $0, map[$1]},每一行数据都会属于某个文件,FILENAME标识出当前行所属的文件名,像在括号{}前的ARGV[1]==FILENAME 你可以看做是其他语法中的条件判断,你可以认为它就是省略了if,但和if的功能是一致的。

    所以上述代码的含义就是 如果当前行是输入name.txt的,就把学号和姓名存在map里(awk里的变量不用预先声明)。 如果当前行是属于score.txt,就从map里把姓名找出来,然后把数据输出。

    结语

    awk 'ARGV[1]==FILENAME {map[$1]=$2} ARGV[2]==FILENAME {print $0, map[$1]}' name.txt score.txt  
    

    针对于不同的数据,只需要调整下后面的具体值,可以实现用不同的列作为key来做join,其实这里并不比sql的join复杂,但在linux服务器上却很方便。上网中我只是实现了right join,如果在`print0, map[1]`前加上`if (length(map[1]) > 0)` 就可以实现inner join。left join的话也只需要把文件名换一下。

    知道了这些,awk实现多文件的交集、差集等操作都不在话下。另外不要忘记了awk其实也是一门编程语言,所以它也可以实现很多很复杂的逻辑,你可以把代码在某个文件里然后用-f参数调起,比如我之前老师用awk做一些简单的统计工作,比如计算均值、总和…… awk简直堪称后端工程师提效利器

    本文来自https://blog.csdn.net/xindoo

    相关文章

      网友评论

          本文标题:awk实现类sql的join操作

          本文链接:https://www.haomeiwen.com/subject/nlhqbktx.html