记录使用List时,获取指定列或者将List展开到一列或者一行的方法
贯彻这个问题和核心在于List结构泛化能力太强,完全可以接受不等长数据存储,然而,在narray和torch.tensor中明确要求为矩阵,也就是明确数据的长和宽。
# 然而在List中,你可能遇到这样的数据
Lista = [
[1, 2, 3]
[1, 2]
[2, 3, 4, 5, 5, 7]
[1]
]
这种情况一定不会在narray和torch.tensor中出现,初始化的时候就会报错。
这也是导致List无法使用列索引的关键原因。
因此对于List如果你能确定你要找的列索引确实都存在,可以通过for循环或者列表表达式的方式提出List的指定列或者将List展开。
# 指定列
a = [
[1, 2]
[1, 3]
[1, 5]
]
a1 = [a_[1] for a_ in a]
# 展开
aa = []
for a_ in a:
aa = aa + a_
注意事项
如果是特别需要使用List处理的矩阵或者大量数据,请进行使用narray或者troch.tensor,像遇到的这种情况,List的处理时间复杂度都是O(n)。很慢。
个人博客:https://kecilimu-notion.vercel.app/
个人邮箱:[email protected]
转载请务必注明出处
文章主图是陌芋大大的图