开发者

insert element in every position of nested list

In Maple 15, I have a nested list like this one (only 2 levels of nesting):

[[a,b],[c],[d,e,f]]

I want to create 6 lists from that one (the number of elements..), like these:

[[x,a,b],[c],[d,e,f]]
[[a,x,b],[c],[d,e,f]]
[[a,b],[x,c],[d,e,f]]
[[a,b],[c],[x,d,e,f]]
[[a,b],[c],[d,x,e,f]]
[[a,b],[c],[d,e,x,f]]

I tried but I have super complex loops and it doesn't work.. 开发者_StackOverflow社区any more brilliant idea?


I am writing this code on Maple 7, but it should work with pretty much any version of Maple. I feel that it is easier to do things by first stripping the original nested list into two simple lists:

input:=[[a,b],[c],[d,e,f]];
flatlist:=[seq(op(input[n]),n=1..nops(input))]; % flat list of all elements
noels:=[seq(nops(input[n]),n=1..nops(input))]; % numbers of elements in sub-lists

It is easier to insert your extra elements into the new flat lists:

newlists:=[seq([seq(flatlist[i],i=1..n-1),x,seq(flatlist[i],i=n..nops(flatlist))],n=1..nops(flatlist))];
newnoels:=[seq(seq([seq(noels[i],i=1..n-1),noels[n]+1,seq(noels[i],i=n+1..nops(noels))],m=1..noels[n]),n=1..nops(noels))];

The resulting flat lists must be converted back into the nested lists. This can be done by writing a simple function:

mergelists:=proc(flatlist,noels)
  local i,j,n,result;
  i:=0; result:=NULL;
  for n from 1 to nops(noels) do
    result:=result,[seq(flatlist[i+j],j=1..noels[n])];
    i:=i+noels[n];
  od;
  return [result];
end:

With this function available, it is now easy to convert the prepared data into the required format:

output:=[seq(mergelists(newlists[n],newnoels[n]),n=1..nops(newlists))];

I admit that this may not be the most elegant solution possible. Unfortunately, nothing simpler comes to my mind just now.


Try a nested pair of seq calls.

L:=[[a,b],[c],[d,e,f]]:

[seq(seq(subs(r=op([x,r]),L),r in K),K in L)];
0

上一篇:

下一篇:

精彩评论

暂无评论...
验证码 换一张
取 消

最新问答

问答排行榜