Julia如何扁平化字符串数组？避免拆分字符串为字符

在Julia中，String本身属于字符的可迭代序列，直接使用Iterators.flatten时会把单个字符串拆分成字符。要实现将单个字符串当作标量处理，同时正常展开字符串数组，需要先对输入元素做预处理：将单个String包装为单元素的可迭代容器，再进行flatten操作。

修改后的实现代码 #

f(xs...) = collect(Iterators.flatten(x isa String ? (x,) : x for x in xs))

测试效果 #

julia> f("oh", ["a", "b"], "uh")
4-element Vector{String}:
 "oh"
 "a"
 "b"
 "uh"

原理说明 #

• 对输入中的每个元素x做判断：
  • 如果是String，就用元组(x,)将其包装成仅包含自身的可迭代对象；
  • 如果是字符串数组这类可迭代容器，则直接保留原对象。
• Iterators.flatten会遍历每个预处理后的可迭代对象，取出其中的元素——对于包装后的字符串，取出的就是完整的String，而非单个字符。

如果需要兼容所有AbstractString类型（比如SubString），可以把判断条件改为x isa AbstractString：

f(xs...) = collect(Iterators.flatten(x isa AbstractString ? (x,) : x for x in xs))

内容的提问来源于stack exchange，提问作者Ryo