切片索引或結束，如果結束較小那麼指數（蟒蛇）

Question

我想獲得切片一些np.ndarray對象foo：

bar = foo[:end] 

但有時最終會比len(foo)更大。然後我想獲得bar = foo。我可以達到這個，如果我寫bar = foo[:min(end, len(foo)]。但它似乎不pythonic。有沒有簡單的方法來做到這一點？

Answer 1

你其實並不需要任何特殊的邏輯來處理的範圍內切出。默認情況下，如果end過大，切片將包括數組的結尾。

>>> a = np.array([1,2,3]) 
>>> a 
array([1, 2, 3]) 
>>> a = a[:10] 
>>> a 
array([1, 2, 3])