说我们有一个清单:片远在每个(名单)第一个元素元素在Python
a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
然后,我要创建通过在切离中的每个元素的第一个元素的新列表清单:
b = a[][1:]
显然,上述不工作,但我想b现在要的是:
[[2, 3], [5, 6], [8, 9]]
当然我可以通过它来循环,但我想知道的是,如果可以以类似的方式来做到这一点,就像我上面做错了那样。另外,最好是有一个更好/好的方式来做到这一点与numpy,在这种情况下,它不需要通过切片我试图做到这一点?
在Python列表和numpy数组之间进行转换不是一个好主意会在这个地方 - 它需要相当多的CPU。你应该尽量减少这些转换。 –
如果我已经将所有内容都放在numpy数组中,是不是指定一个numpy数组? –
不,我的意思是,如果你正在从一个列表转换为一个numpy数组,只是为了完成切片然后再转换回列表。使用列表理解可能会更快,因为即使numpy slice非常快,来回转换速度非常慢。 –