python之根据最小子列表的长度对子列表进行下采样

我有一个像下面这样的列表列表，子列表的数量和长度是可变的:

test = [[1, 5, 4, 3, 5, 2], [4, 2], [5, 2, 4, 3, 5], [5, 3, 1]] 

我想将所有子列表下采样到最短子列表的长度 - 本例 2。这意味着我想从所有子列表中随机选择 2 个元素作为输出。

对于包含大约 100 个子列表的更大列表，每个子列表都超过 100000 个项目，最有效的方法是什么？

请您参考如下方法:

使用 generator expression和 list comprehension与 random.sample()喜欢:

代码:

min_len = min(len(x) for x in data) 
[random.sample(x, min_len) for x in data] 

测试代码:

import random 
 
data = [[1, 5, 4, 3, 5, 2], [4, 2], [5, 2, 4, 3, 5], [5, 3, 1]] 
min_len = min(len(x) for x in data) 
print([random.sample(x, min_len) for x in data]) 

结果:

[[5, 4], [4, 2], [4, 5], [5, 3]]