Python程式：從陣列中獲取指定數量的前幾個元素

---

陣列是一種資料結構，它包含一組相同資料型別的元素，每個元素都由索引標識。

Python中的陣列

Python本身沒有用於表示陣列的資料結構。但是，我們可以使用列表資料結構作為陣列的替代品。這裡我們將使用列表作為陣列。

[10, 4, 11, 76, 99]

Python也提供了一些更合適的模組，例如NumPy和array模組。

使用array模組定義的整數陣列：

array('i', [1, 2, 3, 4])

使用NumPy模組定義的NumPy陣列：

array([1, 2, 3, 4])

在這篇文章中，我們將學習如何從陣列中獲取指定數量的前幾個元素。

輸入輸出場景

假設我們有一個包含9個整數值的輸入陣列。在輸出中，根據指定的數量訪問前幾個元素。

Input array:
[1, 2, 3, 4, 5, 6, 7, 8, 9]
Output:
[1, 2, 3]

從輸入陣列中訪問前3個元素1、2、3。讓我們來看一個包含所有字串元素的陣列。

Input array:
[‘a’, ‘b’, ‘c’, ‘d’, ‘e’]
Output:
[‘a’, ‘b’]

從輸入陣列中檢索前2個元素。在下面的示例中，我們將主要使用Python切片功能來檢索前幾個元素。

切片

切片用於訪問序列中的元素組。以下是執行切片的語法：

sequence_object[start : end]

其中：

• 起始索引 - 切片開始的起始索引。預設為0。

• 結束索引 - 切片停止的結束索引。預設值為可迭代物件的長度。此值被排除在外。

使用列表

我們可以使用列表切片功能來訪問陣列中指定數量的前幾個元素。

示例

讓我們舉個例子，並應用列表切片來訪問指定數量的前幾個元素。

# creating array
lst = [1, 2, 0, 4, 1, 2, 3, 8] 

print ("The original array is: ", lst) 
print() 
numOfItems = 4
# Get first number of elements
result = lst[:numOfItems]
print ("The first {} number of elements are: {}".format(numOfItems, result))

輸出

The original array is:  [1, 2, 0, 4, 1, 2, 3, 8]
The first 4 number of elements are: [1, 2, 0, 4]

使用lst[:numOfItems]語法從給定陣列中訪問前4個元素，並將這些元素儲存在result變數中。

示例

在這個例子中，我們將嘗試從陣列中訪問超出數量的元素。

# creating array
lst = [1, 2, 0] 

print ("The original array is: ", lst) 
print() 
numOfItems = 4

# Get first number of elements
result = lst[:numOfItems]
print ("The first {} number of elements are: {}".format(numOfItems, result))

輸出

The original array is:  [1, 2, 0]
The first 4 number of elements are: [1, 2, 0]

請求的元素數量比陣列列表中可用的元素總數多。切片物件lst[:numOfItems]沒有報錯，而是隻顯示了可用的元素。

使用NumPy陣列

與列表一樣，我們也可以使用NumPy陣列來訪問陣列元素。

示例

在這個例子中，我們將嘗試使用陣列切片功能從NumPy陣列中訪問前2個元素。

import numpy

# creating array
numpy_array = numpy.array([1, 3, 5, 6, 2, 9, 8])

print ("The original array is: ", numpy_array) 
print() 

numOfItems = 2

# get first number of elements
result = numpy_array[:numOfItems]
print ("The result is: ", result) 

輸出

The original array is:  [1 3 5 6 2 9 8]
The result is:  [1 3]

從NumPy陣列物件中訪問前兩個元素1,3。

使用array模組

array模組是Python內建模組，用於使用array()方法定義陣列物件。

示例

在這個例子中，我們將使用array模組建立一個整數陣列。

import array
# creating array
arr = array.array('i', [2, 1, 4, 3, 6, 5, 8, 7])
print ("The original array is: ", arr) 
print() 
numOfItems = 2
# remove first elements
result = arr[:numOfItems]
print ("The result is: ", result) 

輸出

The original array is:  array('i', [2, 1, 4, 3, 6, 5, 8, 7])
The result is:  array('i', [2, 1])

輸入陣列arr中的前2個元素儲存在result變數中。