說我們有一個清單:片遠在每個(名單)第一個元素元素在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非常快,來回轉換速度非常慢。 –