Удалить повторяющиеся элементы списка, сохраняя порядок появления

Я делаю плоские списки с 10^6 до 10^7 действительными числами, и некоторые из них повторяются.

Мне нужно удалить повторяющиеся экземпляры, сохраняя только первое вхождение и не изменяя порядок списка.

Ключевым моментом здесь является эффективность, так как у меня много списков для обработки.

Пример (подделка):

Входные данные:

  {.8, .3 , .8, .5, .3, .6}

Желаемый вывод

  {.8, .3, .5, .6}  

В сторону примечание

Удаление повторяющихся элементов с помощью Union (без сохранения порядка) дает в ноутбуке моего бедняка:

DiscretePlot[a = RandomReal[10, i]; First@Timing@Union@a, {i, 10^6 Range@10}]

3 ответа

Решение

Ты хочешь DeleteDuplicates, который сохраняет порядок списка:

In[13]:= DeleteDuplicates[{.8, .3, .8, .5, .3, .6}]

Out[13]= {0.8, 0.3, 0.5, 0.6}

Это было добавлено в Mathematica 7.0.

Не конкурировать с другими ответами, но я просто не мог не поделиться Compile решение на основе. Решение основано на построении бинарного дерева поиска, а затем проверяет для каждого числа в списке, является ли его индекс в списке тем, который используется при построении b-дерева. Если да, то это исходный номер, если нет - это дубликат. Что делает это решение интересным для меня, так это то, что оно показывает способ эмулировать "передачу по ссылке" Compile, Дело в том, что, если мы встроим скомпилированные функции в другие скомпилированные функции (и это может быть достигнуто с помощью опции "InlineCompiledFunctions"), мы можем ссылаться во внутренних функциях на переменные, определенные во внешней области действия функции (из-за того, как работает встраивание), Это не настоящий переход по ссылке, но он по-прежнему позволяет комбинировать функции из меньших блоков без потери эффективности (это больше в духе макро-расширения). Я не думаю, что это задокументировано, и не знаю, останется ли это в будущих версиях. В любом случае, вот код:

(* A function to build a binary tree *)
Block[{leftchildren , rightchildren},
makeBSearchTree = 
Compile[{{lst, _Real, 1}},
Module[{len = Length[lst], ctr = 1, currentRoot = 1},
 leftchildren = rightchildren =  Table[0, {Length[lst]}];
 For[ctr = 1, ctr <= len, ctr++,
  For[currentRoot = 1, lst[[ctr]] != lst[[currentRoot]],(* 
   nothing *),
   If[lst[[ctr]] < lst[[currentRoot]],
    If[leftchildren[[currentRoot]] == 0,
     leftchildren[[currentRoot]] = ctr;
     Break[],
     (* else *)
     currentRoot = leftchildren[[currentRoot]] ],
    (* else *)
    If[rightchildren[[currentRoot]] == 0,
     rightchildren[[currentRoot]] = ctr;
     Break[],
     (* else *)
     currentRoot = rightchildren[[currentRoot]]]]]];
 ], {{leftchildren, _Integer, 1}, {rightchildren, _Integer, 1}},
CompilationTarget -> "C", "RuntimeOptions" -> "Speed",
CompilationOptions -> {"ExpressionOptimization" -> True}]];


(* A function to query the binary tree and check for a duplicate *)
Block[{leftchildren , rightchildren, lst},
isDuplicate = 
Compile[{{index, _Integer}},
Module[{currentRoot = 1, result = True},
 While[True,
  Which[
   lst[[index]] == lst[[currentRoot]],
    result = index != currentRoot;
    Break[],
   lst[[index]] < lst[[currentRoot]],
    currentRoot = leftchildren[[currentRoot]],
   True,
    currentRoot = rightchildren[[currentRoot]]
   ]];
 result
 ],
{{leftchildren, _Integer, 1}, {rightchildren, _Integer, 
  1}, {lst, _Real, 1}},
CompilationTarget -> "C", "RuntimeOptions" -> "Speed",
CompilationOptions -> {"ExpressionOptimization" -> True}
]];


(* The main function *)
Clear[deldup];
deldup = 
Compile[{{lst, _Real, 1}},
  Module[{len = Length[lst], leftchildren , rightchildren , 
     nodup = Table[0., {Length[lst]}], ndctr = 0, ctr = 1},
makeBSearchTree[lst]; 
For[ctr = 1, ctr <= len, ctr++,
 If[! isDuplicate [ctr],
  ++ndctr;
   nodup[[ndctr]] =  lst[[ctr]]
  ]];
Take[nodup, ndctr]], CompilationTarget -> "C", 
"RuntimeOptions" -> "Speed",
CompilationOptions -> {"ExpressionOptimization" -> True,
 "InlineCompiledFunctions" -> True, 
 "InlineExternalDefinitions" -> True}];

Вот несколько тестов:

In[61]:= intTst = N@RandomInteger[{0,500000},1000000];

In[62]:= (res1 = deldup[intTst ])//Short//Timing
Out[62]= {1.141,{260172.,421188.,487754.,259397.,<<432546>>,154340.,295707.,197588.,119996.}}

In[63]:= (res2 = Tally[intTst,Equal][[All,1]])//Short//Timing
Out[63]= {0.64,{260172.,421188.,487754.,259397.,<<432546>>,154340.,295707.,197588.,119996.}}

In[64]:= res1==res2
Out[64]= True

Не так быстро, как Tally версия, но также Equal - и, как я уже сказал, моей целью было проиллюстрировать интересную (IMO) технику.

Для версий Mathematica до 7 и для общего интереса, вот несколько способов реализации функции UnsortedUnion (то есть DeleteDuplicates). Они собраны из справочных документов и MathGroup. Они были настроены так, чтобы принимать несколько списков, которые затем объединяются, по аналогии с Union.

Для Mathematica 4 или ранее

UnsortedUnion = Module[{f}, f[y_] := (f[y] = Sequence[]; y); f /@ Join@##] &

Для Mathematica 5

UnsortedUnion[x__List] := Reap[Sow[1, Join@x], _, # &][[2]]

Для Mathematica 6

UnsortedUnion[x__List] := Tally[Join@x][[All, 1]]

От Леонида Шифрина для Mathematica 3+ (?)

unsortedUnion[x_List] := Extract[x, Sort[Union[x] /. Dispatch[MapIndexed[Rule, x]]]]
Другие вопросы по тегам