Question

我有两个列表操作，我想请求帮助。我实施它们的方式不是很优雅，所以我想向你们学习专家。

1）假设我有两个列表，一个是{{0,2,4},{1,3,2},{2,0,4}}，另一个是{{1,3,7},{2,4,6},{3,1,9}}。我想要根据值，或者基于某些标准来过滤第一个列表，然后在第二个列表中获取相应的元素。例如，基于非零的值，我想获得{{3,7},{2,4,6},{3,9}}。根据大于2的条件，我想获得{{7},{4},{9}}。

2）我有一个列表，例如{{{1,2},{1,1}},{{1,3},{2,4}},{{1,2},{2,3}},{{1,4},{3,3}}}。我想形成{{{1,2},{{1,1},{2,3}}},{{1,3},{{2,4}}},{{1,4},{{3,3}}}。也就是说，如果第一个元素相同，我想将这些第二个列表分组。我怎么能以漂亮的方式做到这一点？

非常感谢。

Answer 1

对于第一部分，您需要Pick：

In[27]:= Pick[{{1,3,7},{2,4,6},{3,1,9}},{{0,2,4},{1,3,2},{2,0,4}},_?Positive]
Out[27]= {{3,7},{2,4,6},{3,9}}

In[28]:= Pick[{{1,3,7},{2,4,6},{3,1,9}},{{0,2,4},{1,3,2},{2,0,4}},_?(#>2&)]
Out[28]= {{7},{4},{9}}

对于第二个问题，GatherBy可以帮助您完成大部分工作：

In[29]:= x = GatherBy[{{{1, 2}, {1, 1}}, {{1, 3}, {2, 4}}, {{1, 2}, 
    {2, 3}}, {{1, 4}, {3, 3}}}, First]

Out[29]= {{{{1, 2}, {1, 1}}, {{1, 2}, {2, 3}}}, {{{1, 3}, 
    {2, 4}}}, {{{1, 4}, {3, 3}}}}

然后你可以应用一个规则来清理一下：

In[30]:= x /. l:{{a_, _}..} :> {a, Last /@ l}

Out[30]= {{{1, 2}, {{1, 1}, {2, 3}}}, {{1, 3}, {{2, 4}}},
    {{1, 4}, {{3, 3}}}}

Answer 2

正如迈克尔所说，Pick绝对是第一种方式。

对于第二部分，我想提供一个替代方案，让您在一行中执行此操作：SelectEquivalents。（我知道，相当自我推销，但我经常使用这个功能。）为了得到你想要的结果，只需输入

In[1] := SelectEquivalents[ <list>, First, Last, {#1,#2}& ]
Out[1]:= {{{1, 2}, {{1, 1}, {2, 3}}}, {{1, 3}, {{2, 4}}}, {{1, 4}, {{3, 3}}}}

在内部，SelectEquivalents使用Reap和Sow，因此First标记<list>中的每个元素，Last将元素转换为表单我们希望使用，{#1, #2}&会返回包含{Tag, {<items with that tag>}}形式元素的列表。优势在于您可以一步到位地指定所有内容，无需后续转换即可获得所需内容。

mathematica中的两个列表操作

2 个答案: