[Python]配列(Array)の先頭から複数の要素を取得するには？

どうも、ちょげ(@chogetarou)です。

配列の先頭からN個の要素を取得する方法を紹介します。

方法

配列の先頭からN個の要素を取得するには、スライスを使います。

具体的には、配列を「[:N]」(Nは要素数)でスライスします。

array[:N] #リストの先頭からN個を取得

「[:N]」でスライスすれば、配列の先頭からN個の要素を配列として取得できます。

使用例

#リストの場合
numbers = [1, 2, 3, 4, 5]

first = numbers[:3]

print(first) #[1, 2, 3]


#Array.arrayの場合
import array

numbers = array.array('i', [1, 2, 3, -10, 4, 5])

first = numbers[:2]

print(first) #array('i', [1, 2])