上次我們介紹了如何使用 import ,這次我們來介紹 NumPy 函式庫。
它最主要的功能就是能讓我們創建陣列。
首先我們先套用模組。import numpy as np
我們會將 NumPy 簡寫為 np,之後就可以使用「np.」為前綴來調用 NumPy 的功能和函式,若沒有打「as np」,則前綴就必須完整打出「numpy.」。
NumPy 最基本的用法就是用來創建陣列,我們可以使用 np.array() 來創建。
# 一維陣列
array1 = np.array([1, 2, 3, 4, 5, 6])
# 二維陣列
array2 = np.array([[1, 2, 3], [4, 5, 6]])
建立陣列時,我們還可以設定陣列的最小維度和資料型態。array3 = np.array([1, 2, 3, 4, 5, 6], ndmin=2, dtype='float')
ndmin設定把原本一維陣列變成二維陣列,dtype設定資料型態為浮點數
(下篇會介紹 NumPy 的資料型態)。
我們還可以透過「reshape」把一維陣列轉成二維陣列。new_array1 = array1.reshape(2, 3)
NumPy還有很多函式用來創建不同的陣列。
函式 | 說明 |
---|---|
np.array() | 根據串列建立陣列 |
np.empty() | 創建空陣列 |
np.zeros() | 建立數值都為0的陣列 |
np.ones() | 建立數值都為1的陣列 |
np.eye() | 建立對角線數值為1,其他數值為0的陣列 |
np.tile() | 重複現有的陣列內容,創建新維度的陣列 |
這些函式都可以用來創建多維陣列。