q语言如何根据布尔列表取值更直接地合并两个列表
q语言按布尔掩码合并列表的更优实现
我目前有一套可实现该需求的方法,也能借此清晰说明我的问题:
a:1 2 3 4; b:5 6 7; cond:1101001b; comb:(count cond) # 0N; comb[where cond]:a; comb[where not cond]:b
q语言内置了大量列表操作相关的工具函数,我想了解是否存在更直接的方式可以实现该列表合并需求。
解答
存在更直接的原生实现,不需要预先初始化空占位列表,直接使用q的向量条件运算符即可一行完成需求:
comb:cond ?[b;a]
用法说明
- 该写法是q原生向量操作,执行效率高于手动分配索引赋值的实现,无额外性能开销
- 语法逻辑:当
?的左操作数为布尔类型列表时,boolCond ? [valWhenFalse; valWhenTrue]会自动按掩码顺序取值:掩码为0b的位置按顺序取第二个参数(此处为列表b)的元素,掩码为1b的位置按顺序取第三个参数(此处为列表a)的元素 - 输入要求和你当前的实现完全匹配:b的长度等于cond中
0b的个数,a的长度等于cond中1b的个数,运行结果和你给出的原有实现完全一致,返回结果为1 2 5 3 6 7 4
如果需要处理等长列表按位置选值的场景,还可以用逐元素条件判断写法cond?'[a;b],但针对你当前按掩码顺序填充不等长列表的需求,cond ?[b;a]是最简洁的内置方案。
内容的提问来源于stack exchange,提问作者Chechy Levas
相关产品推荐
相关产品推荐

