arrays - 替换索引后的数组项

Question

让

A=[x1,x2,x3,y1,y2,y3,y4]

我想替换数组 A 中 y1 之后的所有内容:

A=[x1,x2,x3,y1,y1,y1,y1]

我试过了

for i in eachindex(A)
   if A[i]==y1
      for j in i:length(A)
        A[j]=y1
      end
   end 
end

但是看起来很复杂，有没有其他的方法可以简单点呢？

Answer 1

这是你想要的吗？在示例中，我假设您的 y1 是 1。

julia> A = [5, 4, 3, 1, 2, 7, 8]
7-element Vector{Int64}:
 5
 4
 3
 1
 2
 7
 8

julia> loc = findfirst(==(1), A)
4

julia> isnothing(A) || (A[loc:end] .= 1);

julia> A
7-element Vector{Int64}:
 5
 4
 3
 1
 1
 1
 1