python - 从元组中切片列的快速方法

Question

我有一个巨大的元组列表，我想从中提取各个列。我尝试过两种方法。假设列表名称的名称是List，我想提取第j列。

第一个是

column=[item[j] for item in List]

第二个是

newList=zip(*List)
column=newList[j]

但是这两种方法都太慢，因为列表的长度约为 50000，每个元组的长度约为 100。是否有更快的方法从列表中提取列？

Answer 1

这是 numpy 擅长的事情

A = np.array(Lst) # this step may take a while now ... maybe you should have Lst as a np.array before you get to this point

sliced = A[:,[j]] # this should be really quite fast

这就是说

newList=zip(*List)
column=newList[j]

对于我来说，使用 50kx100 元组只需要不到一秒的时间...所以也许可以分析您的代码并确保瓶颈实际上位于您认为的位置...