朱莉娅

时间:2020-05-21 03:08:27

标签: dictionary julia elementwise-operations

我想将操作广播到字典的所有值。对于数组,我知道我可以使用以下命令广播按元素操作:

julia> b1 = [1, 2, 3]
julia> b1./2
3-element Array{Float64,1}:
 0.5
 1.0
 1.5

对字典的所有值广播相同操作的有效方法是什么?说,对于字典

a1 = Dict("A"=>1, "B"=>2)

1 个答案:

答案 0 :(得分:5)

为字典的键和值定义了迭代协议,因此您可以这样做,例如:

julia> d = Dict("a"=>1, "b"=>2)
Dict{String,Int64} with 2 entries:
  "b" => 2
  "a" => 1

julia> values(d).^2
2-element Array{Int64,1}:
 4
 1

如果要就地更改字典,请使用map!,例如:

julia> map!(x->x^2, values(d))
Base.ValueIterator for a Dict{String,Int64} with 2 entries. Values:
  4
  1

julia> d
Dict{String,Int64} with 2 entries:
  "b" => 4
  "a" => 1

但是,您的函数必须输出可以转换回字典值类型的类型。在我的示例中,我对Int进行平方,得到Int。但是,在这个问题中,您要除以2,这显然会产生Float64。如果无法将浮点数转换回整数,则会出现错误。

请注意,您也可以通过按键进行广播,例如:

julia> f(x) = "hello mr $(x)"
f (generic function with 1 method)

julia> f.(keys(d))
2-element Array{String,1}:
 "hello mr b"
 "hello mr a"

但这不能 就地完成,即您不能在键上使用map!

重要的是,请注意,您应该实例化集合。确实,这将是低效。因此,请避免使用类似collect(values(d)) ./ 2的构造。