按二维列表的列拆分为新列表

Question

我有UCI数据仓库中的胸部手术数据集
此数据集中的一行如下所示：

DGN3,4.36,3.28,PRZ1,F,F,F,T,F,OC12,T,F,F,T,F,59,T

我有一个脚本可以将这些行读入子列表：
每个子列表都是文件中的一行，包含一些数据转换（大多数情况下，只是将行中的每个元素转换为一些数值；这部分与问题无关）

因此，每个子列表如下所示：

[3.0, 4.36, 3.28, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 12.0, 1.0, 0.0, 0.0, 1.0, 0.0,59.0, 1.0]

现在，我想做的是将这个子列表转换为两个子列表，其中：

• 第一个列表的ith子列表包含所有元素，但原始列表的ith子列表的最后一个元素除外
• 第二个列表的第一个子列表只包含原始列表第一个子列表的最后一个元素。

例如，上面显示的行（子列表）将拆分为两个子列表，如下所示：

[3.0, 4.36, 3.28, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 12.0, 1.0, 0.0, 0.0, 1.0, 0.0,59.0]
[1.0]

现在，我知道我可以很容易地做到这一点，如下所示：

features, classes = [], []
for subl in L:
    features.append(subl[:-1])
    classes.append([subl[-1]])

我想知道有没有更好的办法。是否有一个itertools配方或numpy的某些东西更适合这个？

Answer 1

就处理列表L而言，您所拥有的代码已经是您能做的最好的了。您可以将其缩短为压缩列表理解，使其成为一行，但这并不能降低其复杂性：

>>> L = [[1, 2, 3, 4, 5], [6, 7, 8, 9], [10, 11, 12], [13, 14, 15, 16, 17], [18, 19, 20]]
>>> features, classes = zip(*[(s[:-1], [s[-1]]) for s in L])
>>> features
([1, 2, 3, 4], [6, 7, 8], [10, 11], [13, 14, 15, 16], [18, 19])
>>> classes
([5], [9], [12], [17], [20])

如果流转时长确实是个问题，您应该考虑直接使用数据转换进行这些拆分，这样您就不会生成最终不会使用的中介格式。