回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我需要帮助来优化我的代码</p>
<p>我有一个数据:</p>
<pre><code>data = [
{"ids": [1]},
{"ids": [3, 4]},
{"ids": [1, 2]},
{"ids": [2]},
]
</code></pre>
<p>我需要按ID对其进行分组,而不交叉,因此预期数据应为:</p>
<pre><code>expected = [
[{"ids": [1]}, {"ids": [2]}],
[{"ids": [3, 4]}, {"ids": [1, 2]}],
] # only 2 sublist here
</code></pre>
<p>我要拆分的代码(未优化):</p>
<pre><code>import itertools as it
def _split(
list_of_dicts,
):
splitted_list_of_dicts = []
sub_list = []
while list_of_dicts:
for dct in list_of_dicts:
ids_in_sub_list = set(
it.chain(*[sub_list_el["ids"] for sub_list_el in sub_list]),
)
if not set(dct["ids"]).intersection(ids_in_sub_list):
sub_list.append(dct)
list_of_dicts.remove(dct)
splitted_list_of_dicts.append(sub_list)
sub_list = []
return splitted_list_of_dicts
</code></pre>
<p>我的代码的结果是:</p>
<pre><code>result = [
[{'ids': [1]}, {'ids': [2]}],
[{'ids': [3, 4]}],
[{'ids': [1, 2]}]
] # 3 sublist
</code></pre>
<p>我又得到了一个列表,我试图对其进行优化。
如果你对如何帮助我有任何想法,我会很高兴的,谢谢你的时间</p>
<p>更多示例:</p>
<pre><code>data = [
{"ids": [1]},
{"ids": [3, 4]},
{"ids": [1, 2]},
{"ids": [4]},
{"ids": [3]},
{"ids": [2]},
]
</code></pre>
<p>可分组为两个元素列表:</p>
<pre><code>expected = [
[{'ids': [1]}, {'ids': [4]}, {'ids': [2]}, {'ids': [3]}],
[{'ids': [3, 4]}, {'ids': [1, 2]}],
]
</code></pre>
<p>但现在我得到了全部4:</p>
<pre><code>result = [
[{'ids': [1]}, {'ids': [4]}, {'ids': [2]}],
[{'ids': [3, 4]}],
[{'ids': [1, 2]}],
[{'ids': [3]}]
]
</code></pre>