【轉】Python之Numpy詳細教程

順其自然發表於2021-01-05

 

NumPy - 簡介

NumPy 是一個 Python 包。 它代表 “Numeric Python”。 它是一個由多維陣列物件和用於處理陣列的例程集合組成的庫。

Numeric,即 NumPy 的前身,是由 Jim Hugunin 開發的。 也開發了另一個包 Numarray ,它擁有一些額外的功能。 2005年,Travis Oliphant 通過將 Numarray 的功能整合到 Numeric 包中來建立 NumPy 包。 這個開源專案有很多貢獻者。

NumPy 操作

使用NumPy,開發人員可以執行以下操作:

  • 陣列的算數和邏輯運算。

  • 傅立葉變換和用於圖形操作的例程。

  • 與線性代數有關的操作。 NumPy 擁有線性代數和隨機數生成的內建函式。

NumPy – MatLab 的替代之一

NumPy 通常與 SciPy(Scientific Python)和 Matplotlib(繪相簿)一起使用。 這種組合廣泛用於替代 MatLab,是一個流行的技術計算平臺。 但是,Python 作為 MatLab 的替代方案,現在被視為一種更加現代和完整的程式語言。

NumPy 是開源的,這是它的一個額外的優勢。

 

NumPy - Ndarray 物件

NumPy 中定義的最重要的物件是稱為 ndarray 的 N 維陣列型別。 它描述相同型別的元素集合。 可以使用基於零的索引訪問集合中的專案。

ndarray中的每個元素在記憶體中使用相同大小的塊。 ndarray中的每個元素是資料型別物件的物件(稱為 dtype)。

ndarray物件提取的任何元素(通過切片)由一個陣列標量型別的 Python 物件表示。 下圖顯示了ndarray,資料型別物件(dtype)和陣列標量型別之間的關係。

 

Ndarray

ndarray類的例項可以通過本教程後面描述的不同的陣列建立例程來構造。 基本的ndarray是使用 NumPy 中的陣列函式建立的,如下所示:

 
  1.  
  2. numpy.array

  3.  
  4.  

它從任何暴露陣列介面的物件,或從返回陣列的任何方法建立一個ndarray。

 
  1.  
  2. numpy.array(object, dtype = None, copy = True, order = None, subok = False, ndmin = 0)

  3.  
  4.  

上面的構造器接受以下引數:

序號引數及描述
1.object 任何暴露陣列介面方法的物件都會返回一個陣列或任何(巢狀)序列。
2.dtype 陣列的所需資料型別,可選。
3.copy 可選,預設為true,物件是否被複制。
4.order C(按行)、F(按列)或A(任意,預設)。
5.subok 預設情況下,返回的陣列被強制為基類陣列。 如果為true,則返回子類。
6.ndmin 指定返回陣列的最小維數。

看看下面的例子來更好地理解。

示例 1

 
  1.  
  2. import numpy as np

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

  4. print a

  5.  

輸出如下:

 
  1. [1, 2, 3]

  2.  

示例 2

 
  1.  
  2. # 多於一個維度

  3. import numpy as np

  4. a = np.array([[1, 2], [3, 4]])

  5. print a

  6.  

輸出如下:

 
  1. [[1, 2]

  2. [3, 4]]

  3.  

示例 3

 
  1.  
  2. # 最小維度

  3. import numpy as np

  4. a = np.array([1, 2, 3,4,5], ndmin = 2)

  5. print a

  6.  

輸出如下:

 
  1. [[1, 2, 3, 4, 5]]

  2.  

示例 4

 
  1.  
  2. # dtype 引數

  3. import numpy as np

  4. a = np.array([1, 2, 3], dtype = complex)

  5. print a

  6.  

輸出如下:

 
  1. [ 1.+0.j, 2.+0.j, 3.+0.j]

  2.  

**ndarray ** 物件由計算機記憶體中的一維連續區域組成,帶有將每個元素對映到記憶體塊中某個位置的索引方案。 記憶體塊以按行(C 風格)或按列(FORTRAN 或 MatLab 風格)的方式儲存元素。

NumPy - 資料型別

NumPy 支援比 Python 更多種類的數值型別。 下表顯示了 NumPy 中定義的不同標量資料型別。

序號資料型別及描述
1.bool_ 儲存為一個位元組的布林值(真或假)
2.int_ 預設整數,相當於 C 的long,通常為int32int64
3.intc 相當於 C 的int,通常為int32int64
4.intp 用於索引的整數,相當於 C 的size_t,通常為int32int64
5.int8 位元組(-128 ~ 127)
6.int16 16 位整數(-32768 ~ 32767)
7.int32 32 位整數(-2147483648 ~ 2147483647)
8.int64 64 位整數(-9223372036854775808 ~ 9223372036854775807)
9.uint8 8 位無符號整數(0 ~ 255)
10.uint16 16 位無符號整數(0 ~ 65535)
11.uint32 32 位無符號整數(0 ~ 4294967295)
12.uint64 64 位無符號整數(0 ~ 18446744073709551615)
13.float_ float64的簡寫
14.float16 半精度浮點:符號位,5 位指數,10 位尾數
15.float32 單精度浮點:符號位,8 位指數,23 位尾數
16.float64 雙精度浮點:符號位,11 位指數,52 位尾數
17.complex_ complex128的簡寫
18.complex64 複數,由兩個 32 位浮點表示(實部和虛部)
19.complex128 複數,由兩個 64 位浮點表示(實部和虛部)

NumPy 數字型別是dtype(資料型別)物件的例項,每個物件具有唯一的特徵。 這些型別可以是np.bool_np.float32等。

資料型別物件 (dtype)

資料型別物件描述了對應於陣列的固定記憶體塊的解釋,取決於以下方面:

  • 資料型別(整數、浮點或者 Python 物件)

  • 資料大小

  • 位元組序(小端或大端)

  • 在結構化型別的情況下,欄位的名稱,每個欄位的資料型別,和每個欄位佔用的記憶體塊部分。

  • 如果資料型別是子序列,它的形狀和資料型別。

位元組順序取決於資料型別的字首<>。 <意味著編碼是小端(最小有效位元組儲存在最小地址中)。 >意味著編碼是大端(最大有效位元組儲存在最小地址中)。

dtype可由一下語法構造:

 
  1. numpy.dtype(object, align, copy)

  2.  

引數為:

  • Object:被轉換為資料型別的物件。

  • Align:如果為true,則向欄位新增間隔,使其類似 C 的結構體。

  • Copy ? 生成dtype物件的新副本,如果為flase,結果是內建資料型別物件的引用。

示例 1

 
  1.  
  2. # 使用陣列標量型別

  3. import numpy as np

  4. dt = np.dtype(np.int32)

  5. print dt

  6.  

輸出如下:

 
  1. int32

  2.  

示例 2

 
  1.  
  2. #int8,int16,int32,int64 可替換為等價的字串 'i1','i2','i4',以及其他。

  3. import numpy as np

  4.  
  5. dt = np.dtype('i4')

  6. print dt

  7.  

輸出如下:

 
  1. int32

  2.  

示例 3

 
  1.  
  2. # 使用端記號

  3. import numpy as np

  4. dt = np.dtype('>i4')

  5. print dt

  6.  

輸出如下:

 
  1. >i4

  2.  

下面的例子展示了結構化資料型別的使用。 這裡宣告瞭欄位名稱和相應的標量資料型別。

示例 4

 
  1.  
  2. # 首先建立結構化資料型別。

  3. import numpy as np

  4. dt = np.dtype([('age',np.int8)])

  5. print dt

  6.  

輸出如下:

 
  1. [('age', 'i1')]

  2.  

示例 5

 
  1.  
  2. # 現在將其應用於 ndarray 物件

  3. import numpy as np

  4.  
  5. dt = np.dtype([('age',np.int8)])

  6. a = np.array([(10,),(20,),(30,)], dtype = dt)

  7. print a

  8.  

輸出如下:

 
  1. [(10,) (20,) (30,)]

  2.  

示例 6

 
  1.  
  2. # 檔名稱可用於訪問 age 列的內容

  3. import numpy as np

  4.  
  5. dt = np.dtype([('age',np.int8)])

  6. a = np.array([(10,),(20,),(30,)], dtype = dt)

  7. print a['age']

  8.  

輸出如下:

 
  1. [10 20 30]

  2.  

示例 7

以下示例定義名為 student 的結構化資料型別,其中包含字串欄位name整數字段age浮點欄位marks。 此dtype應用於ndarray物件。

 
  1.  
  2. import numpy as np

  3. student = np.dtype([('name','S20'), ('age', 'i1'), ('marks', 'f4')])

  4. print student

  5.  

輸出如下:

 
  1. [('name', 'S20'), ('age', 'i1'), ('marks', '<f4')])

  2.  

示例 8

 
  1.  
  2. import numpy as np

  3.  
  4. student = np.dtype([('name','S20'), ('age', 'i1'), ('marks', 'f4')])

  5. a = np.array([('abc', 21, 50),('xyz', 18, 75)], dtype = student)

  6. print a

  7.  

輸出如下:

 
  1. [('abc', 21, 50.0), ('xyz', 18, 75.0)]

  2.  

每個內建型別都有一個唯一定義它的字元程式碼:

  • 'b':布林值

  • 'i':符號整數

  • 'u':無符號整數

  • 'f':浮點

  • 'c':複數浮點

  • 'm':時間間隔

  • 'M':日期時間

  • 'O':Python 物件

  • 'S', 'a':位元組串

  • 'U':Unicode

  • 'V':原始資料(void

NumPy - 陣列屬性

這一章中,我們會討論 NumPy 的多種陣列屬性。

ndarray.shape

這一陣列屬性返回一個包含陣列維度的元組,它也可以用於調整陣列大小。

示例 1

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[4,5,6]])

  4. print a.shape

  5.  

輸出如下:

 
  1. (2, 3)

  2.  

示例 2

 
  1.  
  2. # 這會調整陣列大小

  3. import numpy as np

  4.  
  5. a = np.array([[1,2,3],[4,5,6]]) a.shape = (3,2)

  6. print a

  7.  

輸出如下:

 
  1. [[1, 2]

  2. [3, 4]

  3. [5, 6]]

  4.  

示例 3

NumPy 也提供了reshape函式來調整陣列大小。

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[4,5,6]])

  4. b = a.reshape(3,2)

  5. print b

  6.  

輸出如下:

 
  1. [[1, 2]

  2. [3, 4]

  3. [5, 6]]

  4.  

ndarray.ndim

這一陣列屬性返回陣列的維數。

示例 1

 
  1.  
  2. # 等間隔數字的陣列

  3. import numpy as np

  4. a = np.arange(24) print a

  5.  

輸出如下:

 
  1. [0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]

  2.  

示例 2

 
  1.  
  2. # 一維陣列

  3. import numpy as np

  4. a = np.arange(24) a.ndim

  5. # 現在調整其大小

  6. b = a.reshape(2,4,3)

  7. print b

  8. # b 現在擁有三個維度

  9.  

輸出如下:

 
  1. [[[ 0, 1, 2]

  2. [ 3, 4, 5]

  3. [ 6, 7, 8]

  4. [ 9, 10, 11]]

  5. [[12, 13, 14]

  6. [15, 16, 17]

  7. [18, 19, 20]

  8. [21, 22, 23]]]

  9.  

numpy.itemsize

這一陣列屬性返回陣列中每個元素的位元組單位長度。

示例 1

 
  1.  
  2. # 陣列的 dtype 為 int8(一個位元組)

  3. import numpy as np

  4. x = np.array([1,2,3,4,5], dtype = np.int8)

  5. print x.itemsize

  6.  

輸出如下:

 
  1. 1

  2.  

示例 2

 
  1.  
  2. # 陣列的 dtype 現在為 float32(四個位元組)

  3. import numpy as np

  4. x = np.array([1,2,3,4,5], dtype = np.float32)

  5. print x.itemsize

  6.  

輸出如下:

 
  1. 4

  2.  

numpy.flags

ndarray物件擁有以下屬性。這個函式返回了它們的當前值。

序號屬性及描述
1.C_CONTIGUOUS (C) 陣列位於單一的、C 風格的連續區段內
2.F_CONTIGUOUS (F) 陣列位於單一的、Fortran 風格的連續區段內
3.OWNDATA (O) 陣列的記憶體從其它物件處借用
4.WRITEABLE (W) 資料區域可寫入。 將它設定為flase會鎖定資料,使其只讀
5.ALIGNED (A) 資料和任何元素會為硬體適當對齊
6.UPDATEIFCOPY (U) 這個陣列是另一陣列的副本。當這個陣列釋放時,源陣列會由這個陣列中的元素更新

示例

下面的例子展示當前的標誌。

 
  1.  
  2. import numpy as np

  3. x = np.array([1,2,3,4,5])

  4. print x.flags

  5.  

輸出如下:

 
  1.  
  2. C_CONTIGUOUS : True

  3. F_CONTIGUOUS : True

  4. OWNDATA : True

  5. WRITEABLE : True

  6. ALIGNED : True

  7. UPDATEIFCOPY : False

  8.  
  9.  

NumPy - 陣列建立例程

新的ndarray物件可以通過任何下列陣列建立例程或使用低階ndarray建構函式構造。

numpy.empty

它建立指定形狀和dtype的未初始化陣列。 它使用以下建構函式:

 
  1. numpy.empty(shape, dtype = float, order = 'C')

  2.  

構造器接受下列引數:

序號引數及描述
1.Shape 空陣列的形狀,整數或整數元組
2.Dtype 所需的輸出陣列型別,可選
3.Order 'C'為按行的 C 風格陣列,'F'為按列的 Fortran 風格陣列

示例

下面的程式碼展示空陣列的例子:

 
  1.  
  2. import numpy as np

  3. x = np.empty([3,2], dtype = int)

  4. print x

  5.  

輸出如下:

 
  1. [[22649312 1701344351]

  2. [1818321759 1885959276]

  3. [16779776 156368896]]

  4.  

注意:陣列元素為隨機值,因為它們未初始化。

numpy.zeros

返回特定大小,以 0 填充的新陣列。

 
  1. numpy.zeros(shape, dtype = float, order = 'C')

  2.  

構造器接受下列引數:

序號引數及描述
1.Shape 空陣列的形狀,整數或整數元組
2.Dtype 所需的輸出陣列型別,可選
3.Order 'C'為按行的 C 風格陣列,'F'為按列的 Fortran 風格陣列

示例 1

 
  1.  
  2. # 含有 5 個 0 的陣列,預設型別為 float

  3. import numpy as np

  4. x = np.zeros(5)

  5. print x

  6.  

輸出如下:

 
  1. [ 0. 0. 0. 0. 0.]

  2.  

示例 2

 
  1.  
  2. import numpy as np

  3. x = np.zeros((5,), dtype = np.int)

  4. print x

  5.  

輸出如下:

 
  1. [0 0 0 0 0]

  2.  

示例 3

 
  1.  
  2. # 自定義型別

  3. import numpy as np

  4. x = np.zeros((2,2), dtype = [('x', 'i4'), ('y', 'i4')])

  5. print x

  6.  

輸出如下:

 
  1. [[(0,0)(0,0)]

  2. [(0,0)(0,0)]]

  3.  

numpy.ones

返回特定大小,以 1 填充的新陣列。

 
  1.  
  2. numpy.ones(shape, dtype = None, order = 'C')

  3.  
  4.  

構造器接受下列引數:

序號引數及描述
1.Shape 空陣列的形狀,整數或整數元組
2.Dtype 所需的輸出陣列型別,可選
3.Order 'C'為按行的 C 風格陣列,'F'為按列的 Fortran 風格陣列

示例 1

 
  1.  
  2. # 含有 5 個 1 的陣列,預設型別為 float

  3. import numpy as np

  4. x = np.ones(5) print x

  5.  

輸出如下:

 
  1. [ 1. 1. 1. 1. 1.]

  2.  

示例 2

 
  1.  
  2. import numpy as np

  3. x = np.ones([2,2], dtype = int)

  4. print x

  5.  

輸出如下:

 
  1. [[1 1]

  2. [1 1]]

  3.  

NumPy - 來自現有資料的陣列

這一章中,我們會討論如何從現有資料建立陣列。

numpy.asarray

此函式類似於numpy.array,除了它有較少的引數。 這個例程對於將 Python 序列轉換為ndarray非常有用。

 
  1.  
  2. numpy.asarray(a, dtype = None, order = None)

  3.  
  4.  

構造器接受下列引數:

序號引數及描述
1.a 任意形式的輸入引數,比如列表、列表的元組、元組、元組的元組、元組的列表
2.dtype 通常,輸入資料的型別會應用到返回的ndarray
3.order 'C'為按行的 C 風格陣列,'F'為按列的 Fortran 風格陣列

下面的例子展示瞭如何使用asarray函式:

示例 1

 
  1.  
  2. # 將列表轉換為 ndarray

  3. import numpy as np

  4.  
  5. x = [1,2,3]

  6. a = np.asarray(x)

  7. print a

  8.  

輸出如下:

 
  1. [1 2 3]

  2.  

示例 2

 
  1.  
  2. # 設定了 dtype

  3. import numpy as np

  4.  
  5. x = [1,2,3]

  6. a = np.asarray(x, dtype = float)

  7. print a

  8.  

輸出如下:

 
  1. [ 1. 2. 3.]

  2.  

示例 3

 
  1.  
  2. # 來自元組的 ndarray

  3. import numpy as np

  4.  
  5. x = (1,2,3)

  6. a = np.asarray(x)

  7. print a

  8.  

輸出如下:

 
  1. [1 2 3]

  2.  

示例 4

 
  1.  
  2. # 來自元組列表的 ndarray

  3. import numpy as np

  4.  
  5. x = [(1,2,3),(4,5)]

  6. a = np.asarray(x)

  7. print a

  8.  

輸出如下:

 
  1. [(1, 2, 3) (4, 5)]

  2.  

numpy.frombuffer

此函式將緩衝區解釋為一維陣列。 暴露緩衝區介面的任何物件都用作引數來返回ndarray

 
  1. numpy.frombuffer(buffer, dtype = float, count = -1, offset = 0)

  2.  

構造器接受下列引數:

序號引數及描述
1.buffer 任何暴露緩衝區藉口的物件
2.dtype 返回陣列的資料型別,預設為float
3.count 需要讀取的資料數量,預設為-1,讀取所有資料
4.offset 需要讀取的起始位置,預設為0

示例

下面的例子展示了frombuffer函式的用法。

 
  1.  
  2. import numpy as np

  3. s = 'Hello World'

  4. a = np.frombuffer(s, dtype = 'S1')

  5. print a

  6.  

輸出如下:

 
  1. ['H' 'e' 'l' 'l' 'o' ' ' 'W' 'o' 'r' 'l' 'd']

  2.  

numpy.fromiter

此函式從任何可迭代物件構建一個ndarray物件,返回一個新的一維陣列。

 
  1. numpy.fromiter(iterable, dtype, count = -1)

  2.  

構造器接受下列引數:

序號引數及描述
1.iterable 任何可迭代物件
2.dtype 返回陣列的資料型別
3.count 需要讀取的資料數量,預設為-1,讀取所有資料

以下示例展示瞭如何使用內建的range()函式返回列表物件。 此列表的迭代器用於形成ndarray物件。

示例 1

 
  1.  
  2. # 使用 range 函式建立列表物件

  3. import numpy as np

  4. list = range(5)

  5. print list

  6.  

輸出如下:

 
  1. [0, 1, 2, 3, 4]

  2.  

示例 2

 
  1.  
  2. # 從列表中獲得迭代器

  3. import numpy as np

  4. list = range(5)

  5. it = iter(list)

  6. # 使用迭代器建立 ndarray

  7. x = np.fromiter(it, dtype = float)

  8. print x

  9.  

輸出如下:

 
  1. [0. 1. 2. 3. 4.]

  2.  

NumPy - 來自數值範圍的陣列

這一章中,我們會學到如何從數值範圍建立陣列。

numpy.arange

這個函式返回ndarray物件,包含給定範圍內的等間隔值。

 
  1.  
  2. numpy.arange(start, stop, step, dtype)

  3.  
  4.  

構造器接受下列引數:

序號引數及描述
1.start 範圍的起始值,預設為0
2.stop 範圍的終止值(不包含)
3.step 兩個值的間隔,預設為1
4.dtype 返回ndarray的資料型別,如果沒有提供,則會使用輸入資料的型別。

下面的例子展示瞭如何使用該函式:

示例 1

 
  1.  
  2. import numpy as np

  3. x = np.arange(5)

  4. print x

  5.  

輸出如下:

 
  1. [0 1 2 3 4]

  2.  

示例 2

 
  1.  
  2. import numpy as np

  3. # 設定了 dtype

  4. x = np.arange(5, dtype = float)

  5. print x

  6.  

輸出如下:

 
  1. [0. 1. 2. 3. 4.]

  2.  

示例 3

 
  1.  
  2. # 設定了起始值和終止值引數

  3. import numpy as np

  4. x = np.arange(10,20,2)

  5. print x

  6.  

輸出如下:

 
  1. [10 12 14 16 18]

  2.  

numpy.linspace

此函式類似於arange()函式。 在此函式中,指定了範圍之間的均勻間隔數量,而不是步長。 此函式的用法如下。

 
  1.  
  2. numpy.linspace(start, stop, num, endpoint, retstep, dtype)

  3.  
  4.  

構造器接受下列引數:

序號引數及描述
1.start 序列的起始值
2.stop 序列的終止值,如果endpointtrue,該值包含於序列中
3.num 要生成的等間隔樣例數量,預設為50
4.endpoint 序列中是否包含stop值,預設為ture
5.retstep 如果為true,返回樣例,以及連續數字之間的步長
6.dtype 輸出ndarray的資料型別

下面的例子展示了linspace函式的用法。

示例 1

 
  1.  
  2. import numpy as np

  3. x = np.linspace(10,20,5)

  4. print x

  5.  

輸出如下:

 
  1. [10. 12.5 15. 17.5 20.]

  2.  

示例 2

 
  1.  
  2. # 將 endpoint 設為 false

  3. import numpy as np

  4. x = np.linspace(10,20, 5, endpoint = False)

  5. print x

  6.  

輸出如下:

 
  1. [10. 12. 14. 16. 18.]

  2.  

示例 3

 
  1.  
  2. # 輸出 retstep 值

  3. import numpy as np

  4.  
  5. x = np.linspace(1,2,5, retstep = True)

  6. print x

  7. # 這裡的 retstep 為 0.25

  8.  

輸出如下:

 
  1.  
  2. (array([ 1. , 1.25, 1.5 , 1.75, 2. ]), 0.25)

  3.  
  4.  

numpy.logspace

此函式返回一個ndarray物件,其中包含在對數刻度上均勻分佈的數字。 刻度的開始和結束端點是某個底數的冪,通常為 10。

 
  1.  
  2. numpy.logscale(start, stop, num, endpoint, base, dtype)

  3.  
  4.  

logspace函式的輸出由以下引數決定:

序號引數及描述
1.start 起始值是base ** start
2.stop 終止值是base ** stop
3.num 範圍內的數值數量,預設為50
4.endpoint 如果為true,終止值包含在輸出陣列當中
5.base 對數空間的底數,預設為10
6.dtype 輸出陣列的資料型別,如果沒有提供,則取決於其它引數

下面的例子展示了logspace函式的用法。

示例 1

 
  1.  
  2. import numpy as np

  3. # 預設底數是 10

  4. a = np.logspace(1.0, 2.0, num = 10)

  5. print a

  6.  

輸出如下:

 
  1. [ 10. 12.91549665 16.68100537 21.5443469 27.82559402

  2. 35.93813664 46.41588834 59.94842503 77.42636827 100. ]

  3.  

示例 2

 
  1.  
  2. # 將對數空間的底數設定為 2

  3. import numpy as np

  4. a = np.logspace(1,10,num = 10, base = 2)

  5. print a

  6.  

輸出如下:

 
  1. [ 2. 4. 8. 16. 32. 64. 128. 256. 512. 1024.]

  2.  

NumPy - 切片和索引

ndarray物件的內容可以通過索引或切片來訪問和修改,就像 Python 的內建容器物件一樣。

如前所述,ndarray物件中的元素遵循基於零的索引。 有三種可用的索引方法型別: 欄位訪問,基本切片高階索引

基本切片是 Python 中基本切片概念到 n 維的擴充套件。 通過將startstopstep引數提供給內建的slice函式來構造一個 Python slice物件。 此slice物件被傳遞給陣列來提取陣列的一部分。

示例 1

 
  1.  
  2. import numpy as np

  3. a = np.arange(10)

  4. s = slice(2,7,2)

  5. print a[s]

  6.  

輸出如下:

 
  1. [2 4 6]

  2.  

在上面的例子中,ndarray物件由arange()函式建立。 然後,分別用起始,終止和步長值272定義切片物件。 當這個切片物件傳遞給ndarray時,會對它的一部分進行切片,從索引27,步長為2

通過將由冒號分隔的切片引數(start:stop:step)直接提供給ndarray物件,也可以獲得相同的結果。

示例 2

 
  1.  
  2. import numpy as np

  3. a = np.arange(10)

  4. b = a[2:7:2]

  5. print b

  6.  

輸出如下:

 
  1. [2 4 6]

  2.  

如果只輸入一個引數,則將返回與索引對應的單個專案。 如果使用a:,則從該索引向後的所有專案將被提取。 如果使用兩個引數(以:分隔),則對兩個索引(不包括停止索引)之間的元素以預設步驟進行切片。

示例 3

 
  1.  
  2. # 對單個元素進行切片

  3. import numpy as np

  4.  
  5. a = np.arange(10)

  6. b = a[5]

  7. print b

  8.  

輸出如下:

 
  1. 5

  2.  

示例 4

 
  1.  
  2. # 對始於索引的元素進行切片

  3. import numpy as np

  4. a = np.arange(10)

  5. print a[2:]

  6.  

輸出如下:

 
  1. [2 3 4 5 6 7 8 9]

  2.  

示例 5

 
  1.  
  2. # 對索引之間的元素進行切片

  3. import numpy as np

  4. a = np.arange(10)

  5. print a[2:5]

  6.  

輸出如下:

 
  1. [2 3 4]

  2.  

上面的描述也可用於多維ndarray

示例 6

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[3,4,5],[4,5,6]])

  4. print a

  5. # 對始於索引的元素進行切片

  6. print '現在我們從索引 a[1:] 開始對陣列切片'

  7. print a[1:]

  8.  

輸出如下:

 
  1.  
  2. [[1 2 3]

  3. [3 4 5]

  4. [4 5 6]]

  5.  
  6. 現在我們從索引 a[1:] 開始對陣列切片

  7. [[3 4 5]

  8. [4 5 6]]

  9.  
  10.  

切片還可以包括省略號(...),來使選擇元組的長度與陣列的維度相同。 如果在行位置使用省略號,它將返回包含行中元素的ndarray

示例 7

 
  1.  
  2. # 最開始的陣列

  3. import numpy as np

  4. a = np.array([[1,2,3],[3,4,5],[4,5,6]])

  5. print '我們的陣列是:'

  6. print a

  7. print '\n'

  8. # 這會返回第二列元素的陣列:

  9. print '第二列的元素是:'

  10. print a[...,1]

  11. print '\n'

  12. # 現在我們從第二行切片所有元素:

  13. print '第二行的元素是:'

  14. print a[1,...]

  15. print '\n'

  16. # 現在我們從第二列向後切片所有元素:

  17. print '第二列及其剩餘元素是:'

  18. print a[...,1:]

  19.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[1 2 3]

  4. [3 4 5]

  5. [4 5 6]]

  6.  
  7. 第二列的元素是:

  8. [2 4 5]

  9.  
  10. 第二行的元素是:

  11. [3 4 5]

  12.  
  13. 第二列及其剩餘元素是:

  14. [[2 3]

  15. [4 5]

  16. [5 6]]

  17.  
  18.  

NumPy - 高階索引

如果一個ndarray是非元組序列,資料型別為整數或布林值的ndarray,或者至少一個元素為序列物件的元組,我們就能夠用它來索引ndarray。高階索引始終返回資料的副本。 與此相反,切片只提供了一個檢視。

有兩種型別的高階索引:整數和布林值。

整數索引

這種機制有助於基於 N 維索引來獲取陣列中任意元素。 每個整數陣列表示該維度的下標值。 當索引的元素個數就是目標ndarray的維度時,會變得相當直接。

以下示例獲取了ndarray物件中每一行指定列的一個元素。 因此,行索引包含所有行號,列索引指定要選擇的元素。

示例 1

 
  1.  
  2. import numpy as np

  3.  
  4. x = np.array([[1, 2], [3, 4], [5, 6]])

  5. y = x[[0,1,2], [0,1,0]]

  6. print y

  7.  

輸出如下:

 
  1. [1 4 5]

  2.  

該結果包括陣列中(0,0)(1,1)(2,0)位置處的元素。

下面的示例獲取了 4X3 陣列中的每個角處的元素。 行索引是[0,0][3,3],而列索引是[0,2][0,2]

示例 2

 
  1.  
  2. import numpy as np

  3. x = np.array([[ 0, 1, 2],[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])

  4. print '我們的陣列是:'

  5. print x

  6. print '\n'

  7. rows = np.array([[0,0],[3,3]])

  8. cols = np.array([[0,2],[0,2]])

  9. y = x[rows,cols]

  10. print '這個陣列的每個角處的元素是:'

  11. print y

  12.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[ 0 1 2]

  4. [ 3 4 5]

  5. [ 6 7 8]

  6. [ 9 10 11]]

  7.  
  8. 這個陣列的每個角處的元素是:

  9. [[ 0 2]

  10. [ 9 11]]

  11.  
  12.  

返回的結果是包含每個角元素的ndarray物件。

高階和基本索引可以通過使用切片:或省略號...與索引陣列組合。 以下示例使用slice作為列索引和高階索引。 當切片用於兩者時,結果是相同的。 但高階索引會導致複製,並且可能有不同的記憶體佈局。

示例 3

 
  1.  
  2. import numpy as np

  3. x = np.array([[ 0, 1, 2],[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])

  4. print '我們的陣列是:'

  5. print x

  6. print '\n'

  7. # 切片

  8. z = x[1:4,1:3]

  9. print '切片之後,我們的陣列變為:'

  10. print z

  11. print '\n'

  12. # 對列使用高階索引

  13. y = x[1:4,[1,2]]

  14. print '對列使用高階索引來切片:'

  15. print y

  16.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[ 0 1 2]

  4. [ 3 4 5]

  5. [ 6 7 8]

  6. [ 9 10 11]]

  7.  
  8. 切片之後,我們的陣列變為:

  9. [[ 4 5]

  10. [ 7 8]

  11. [10 11]]

  12.  
  13. 對列使用高階索引來切片:

  14. [[ 4 5]

  15. [ 7 8]

  16. [10 11]]

  17.  
  18.  

布林索引

當結果物件是布林運算(例如比較運算子)的結果時,將使用此型別的高階索引。

示例 1

這個例子中,大於 5 的元素會作為布林索引的結果返回。

 
  1.  
  2. import numpy as np

  3. x = np.array([[ 0, 1, 2],[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])

  4. print '我們的陣列是:'

  5. print x

  6. print '\n'

  7. # 現在我們會列印出大於 5 的元素

  8. print '大於 5 的元素是:'

  9. print x[x > 5]

  10.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[ 0 1 2]

  4. [ 3 4 5]

  5. [ 6 7 8]

  6. [ 9 10 11]]

  7.  
  8. 大於 5 的元素是:

  9. [ 6 7 8 9 10 11]

  10.  
  11.  

示例 2

這個例子使用了~(取補運算子)來過濾NaN

 
  1.  
  2. import numpy as np

  3. a = np.array([np.nan, 1,2,np.nan,3,4,5])

  4. print a[~np.isnan(a)]

  5.  

輸出如下:

 
  1. [ 1. 2. 3. 4. 5.]

  2.  

示例 3

以下示例顯示如何從陣列中過濾掉非複數元素。

 
  1.  
  2. import numpy as np

  3. a = np.array([1, 2+6j, 5, 3.5+5j])

  4. print a[np.iscomplex(a)]

  5.  

輸出如下:

 
  1. [2.0+6.j 3.5+5.j]

  2.  

NumPy - 廣播

術語廣播是指 NumPy 在算術運算期間處理不同形狀的陣列的能力。 對陣列的算術運算通常在相應的元素上進行。 如果兩個陣列具有完全相同的形狀,則這些操作被無縫執行。

示例 1

 
  1.  
  2. import numpy as np

  3.  
  4. a = np.array([1,2,3,4])

  5. b = np.array([10,20,30,40])

  6. c = a * b

  7. print c

  8.  

輸出如下:

 
  1. [10 40 90 160]

  2.  

如果兩個陣列的維數不相同,則元素到元素的操作是不可能的。 然而,在 NumPy 中仍然可以對形狀不相似的陣列進行操作,因為它擁有廣播功能。 較小的陣列會廣播到較大陣列的大小,以便使它們的形狀可相容。

如果滿足以下規則,可以進行廣播:

  • ndim較小的陣列會在前面追加一個長度為 1 的維度。

  • 輸出陣列的每個維度的大小是輸入陣列該維度大小的最大值。

  • 如果輸入在每個維度中的大小與輸出大小匹配,或其值正好為 1,則在計算中可它。

  • 如果輸入的某個維度大小為 1,則該維度中的第一個資料元素將用於該維度的所有計算。

如果上述規則產生有效結果,並且滿足以下條件之一,那麼陣列被稱為可廣播的

  • 陣列擁有相同形狀。

  • 陣列擁有相同的維數,每個維度擁有相同長度,或者長度為 1。

  • 陣列擁有極少的維度,可以在其前面追加長度為 1 的維度,使上述條件成立。

下面的例稱展示了廣播的示例。

示例 2

 
  1.  
  2. import numpy as np

  3. a = np.array([[0.0,0.0,0.0],[10.0,10.0,10.0],[20.0,20.0,20.0],[30.0,30.0,30.0]])

  4. b = np.array([1.0,2.0,3.0])

  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. print '第二個陣列:'

  9. print b

  10. print '\n'

  11. print '第一個陣列加第二個陣列:'

  12. print a + b

  13.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[ 0. 0. 0.]

  4. [ 10. 10. 10.]

  5. [ 20. 20. 20.]

  6. [ 30. 30. 30.]]

  7.  
  8. 第二個陣列:

  9. [ 1. 2. 3.]

  10.  
  11. 第一個陣列加第二個陣列:

  12. [[ 1. 2. 3.]

  13. [ 11. 12. 13.]

  14. [ 21. 22. 23.]

  15. [ 31. 32. 33.]]

  16.  
  17.  

下面的圖片展示了陣列b如何通過廣播來與陣列a相容。

 

array

NumPy - 陣列上的迭代

NumPy 包包含一個迭代器物件numpy.nditer。 它是一個有效的多維迭代器物件,可以用於在陣列上進行迭代。 陣列的每個元素可使用 Python 的標準Iterator介面來訪問。

讓我們使用arange()函式建立一個 3X4 陣列,並使用nditer對它進行迭代。

示例 1

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a print '\n'

  7. print '修改後的陣列是:'

  8. for x in np.nditer(a):

  9. print x,

  10.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 修改後的陣列是:

  8. 0 5 10 15 20 25 30 35 40 45 50 55

  9.  
  10.  

示例 2

迭代的順序匹配陣列的內容佈局,而不考慮特定的排序。 這可以通過迭代上述陣列的轉置來看到。

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a

  7. print '\n'

  8. print '原始陣列的轉置是:'

  9. b = a.T

  10. print b

  11. print '\n'

  12. print '修改後的陣列是:'

  13. for x in np.nditer(b):

  14. print x,

  15.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 原始陣列的轉置是:

  8. [[ 0 20 40]

  9. [ 5 25 45]

  10. [10 30 50]

  11. [15 35 55]]

  12.  
  13. 修改後的陣列是:

  14. 0 5 10 15 20 25 30 35 40 45 50 55

  15.  
  16.  

迭代順序

如果相同元素使用 F 風格順序儲存,則迭代器選擇以更有效的方式對陣列進行迭代。

示例 1

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a print '\n'

  7. print '原始陣列的轉置是:'

  8. b = a.T

  9. print b

  10. print '\n'

  11. print '以 C 風格順序排序:'

  12. c = b.copy(order='C')

  13. print c for x in np.nditer(c):

  14. print x,

  15. print '\n'

  16. print '以 F 風格順序排序:'

  17. c = b.copy(order='F')

  18. print c

  19. for x in np.nditer(c):

  20. print x,

  21.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 原始陣列的轉置是:

  8. [[ 0 20 40]

  9. [ 5 25 45]

  10. [10 30 50]

  11. [15 35 55]]

  12.  
  13. 以 C 風格順序排序:

  14. [[ 0 20 40]

  15. [ 5 25 45]

  16. [10 30 50]

  17. [15 35 55]]

  18. 0 20 40 5 25 45 10 30 50 15 35 55

  19.  
  20. 以 F 風格順序排序:

  21. [[ 0 20 40]

  22. [ 5 25 45]

  23. [10 30 50]

  24. [15 35 55]]

  25. 0 5 10 15 20 25 30 35 40 45 50 55

  26.  
  27.  

示例 2

可以通過顯式提醒,來強制nditer物件使用某種順序:

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a

  7. print '\n'

  8. print '以 C 風格順序排序:'

  9. for x in np.nditer(a, order = 'C'):

  10. print x,

  11. print '\n'

  12. print '以 F 風格順序排序:'

  13. for x in np.nditer(a, order = 'F'):

  14. print x,

  15.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 以 C 風格順序排序:

  8. 0 5 10 15 20 25 30 35 40 45 50 55

  9.  
  10. 以 F 風格順序排序:

  11. 0 20 40 5 25 45 10 30 50 15 35 55

  12.  
  13.  

修改陣列的值

nditer物件有另一個可選引數op_flags。 其預設值為只讀,但可以設定為讀寫或只寫模式。 這將允許使用此迭代器修改陣列元素。

示例

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a

  7. print '\n'

  8. for x in np.nditer(a, op_flags=['readwrite']):

  9. x[...]=2*x

  10. print '修改後的陣列是:'

  11. print a

  12.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 修改後的陣列是:

  8. [[ 0 10 20 30]

  9. [ 40 50 60 70]

  10. [ 80 90 100 110]]

  11.  
  12.  

外部迴圈

nditer類的構造器擁有flags引數,它可以接受下列值:

序號引數及描述
1.c_index 可以跟蹤 C 順序的索引
2.f_index 可以跟蹤 Fortran 順序的索引
3.multi-index 每次迭代可以跟蹤一種索引型別
4.external_loop 給出的值是具有多個值的一維陣列,而不是零維陣列

示例

在下面的示例中,迭代器遍歷對應於每列的一維陣列。

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '原始陣列是:'

  6. print a

  7. print '\n'

  8. print '修改後的陣列是:'

  9. for x in np.nditer(a, flags = ['external_loop'], order = 'F'):

  10. print x,

  11.  

輸出如下:

 
  1.  
  2. 原始陣列是:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 修改後的陣列是:

  8. [ 0 20 40] [ 5 25 45] [10 30 50] [15 35 55]

  9.  
  10.  

廣播迭代

如果兩個陣列是可廣播的nditer組合物件能夠同時迭代它們。 假設陣列a具有維度 3X4,並且存在維度為 1X4 的另一個陣列b,則使用以下型別的迭代器(陣列b被廣播到a的大小)。

示例

 
  1.  
  2. import numpy as np

  3. a = np.arange(0,60,5)

  4. a = a.reshape(3,4)

  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. print '第二個陣列:'

  9. b = np.array([1, 2, 3, 4], dtype = int)

  10. print b

  11. print '\n'

  12. print '修改後的陣列是:'

  13. for x,y in np.nditer([a,b]):

  14. print "%d:%d" % (x,y),

  15.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[ 0 5 10 15]

  4. [20 25 30 35]

  5. [40 45 50 55]]

  6.  
  7. 第二個陣列:

  8. [1 2 3 4]

  9.  
  10. 修改後的陣列是:

  11. 0:1 5:2 10:3 15:4 20:1 25:2 30:3 35:4 40:1 45:2 50:3 55:4

  12.  
  13.  

NumPy - 陣列操作

NumPy包中有幾個例程用於處理ndarray物件中的元素。 它們可以分為以下型別:

修改形狀

序號形狀及描述
1.reshape 不改變資料的條件下修改形狀
2.flat 陣列上的一維迭代器
3.flatten 返回摺疊為一維的陣列副本
4.ravel 返回連續的展開陣列

numpy.reshape

這個函式在不改變資料的條件下修改形狀,它接受如下引數:

numpy.reshape(arr, newshape, order')

其中:

  • arr:要修改形狀的陣列
  • newshape:整數或者整數陣列,新的形狀應當相容原有形狀
  • order'C'為 C 風格順序,'F'為 F 風格順序,'A'為保留原順序。

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(8)

  4. print '原始陣列:'

  5. print a

  6. print '\n'

  7.  
  8. b = a.reshape(4,2)

  9. print '修改後的陣列:'

  10. print b

  11.  

輸出如下:

 
  1.  
  2. 原始陣列:

  3. [0 1 2 3 4 5 6 7]

  4.  
  5. 修改後的陣列:

  6. [[0 1]

  7. [2 3]

  8. [4 5]

  9. [6 7]]

  10.  

numpy.ndarray.flat

該函式返回陣列上的一維迭代器,行為類似 Python 內建的迭代器。

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(8).reshape(2,4)

  4. print '原始陣列:'

  5. print a

  6. print '\n'

  7.  
  8. print '呼叫 flat 函式之後:'

  9. # 返回展開陣列中的下標的對應元素

  10. print a.flat[5]

  11.  

輸出如下:

 
  1.  
  2. 原始陣列:

  3. [[0 1 2 3]

  4. [4 5 6 7]]

  5.  
  6. 呼叫 flat 函式之後:

  7. 5

  8.  

numpy.ndarray.flatten

該函式返回摺疊為一維的陣列副本,函式接受下列引數:

 
  1.  
  2. ndarray.flatten(order)

  3.  

其中:

  • order'C' -- 按行,'F' -- 按列,'A' -- 原順序,'k' -- 元素在記憶體中的出現順序。

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(8).reshape(2,4)

  4.  
  5. print '原陣列:'

  6. print a

  7. print '\n'

  8. # default is column-major

  9.  
  10. print '展開的陣列:'

  11. print a.flatten()

  12. print '\n'

  13.  
  14. print '以 F 風格順序展開的陣列:'

  15. print a.flatten(order = 'F')

  16.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [[0 1 2 3]

  4. [4 5 6 7]]

  5.  
  6. 展開的陣列:

  7. [0 1 2 3 4 5 6 7]

  8.  
  9. 以 F 風格順序展開的陣列:

  10. [0 4 1 5 2 6 3 7]

  11.  

numpy.ravel

這個函式返回展開的一維陣列,並且按需生成副本。返回的陣列和輸入陣列擁有相同資料型別。這個函式接受兩個引數。

 
  1.  
  2. numpy.ravel(a, order)

  3.  

構造器接受下列引數:

  • order'C' -- 按行,'F' -- 按列,'A' -- 原順序,'k' -- 元素在記憶體中的出現順序。

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(8).reshape(2,4)

  4.  
  5. print '原陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '呼叫 ravel 函式之後:'

  10. print a.ravel()

  11. print '\n'

  12.  
  13. print '以 F 風格順序呼叫 ravel 函式之後:'

  14. print a.ravel(order = 'F')

  15.  
 
  1.  
  2. 原陣列:

  3. [[0 1 2 3]

  4. [4 5 6 7]]

  5.  
  6. 呼叫 ravel 函式之後:

  7. [0 1 2 3 4 5 6 7]

  8.  
  9. 以 F 風格順序呼叫 ravel 函式之後:

  10. [0 4 1 5 2 6 3 7]

  11.  

翻轉操作

序號操作及描述
1.transpose 翻轉陣列的維度
2.ndarray.T 和self.transpose()相同
3.rollaxis 向後滾動指定的軸
4.swapaxes 互換陣列的兩個軸

numpy.transpose

這個函式翻轉給定陣列的維度。如果可能的話它會返回一個檢視。函式接受下列引數:

 
  1.  
  2. numpy.transpose(arr, axes)

  3.  

其中:

  • arr:要轉置的陣列
  • axes:整數的列表,對應維度,通常所有維度都會翻轉。

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(12).reshape(3,4)

  4.  
  5. print '原陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '轉置陣列:'

  10. print np.transpose(a)

  11.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [[ 0 1 2 3]

  4. [ 4 5 6 7]

  5. [ 8 9 10 11]]

  6.  
  7. 轉置陣列:

  8. [[ 0 4 8]

  9. [ 1 5 9]

  10. [ 2 6 10]

  11. [ 3 7 11]]

  12.  

numpy.ndarray.T

該函式屬於ndarray類,行為類似於numpy.transpose

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(12).reshape(3,4)

  4.  
  5. print '原陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '轉置陣列:'

  10. print a.T

  11.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [[ 0 1 2 3]

  4. [ 4 5 6 7]

  5. [ 8 9 10 11]]

  6.  
  7. 轉置陣列:

  8. [[ 0 4 8]

  9. [ 1 5 9]

  10. [ 2 6 10]

  11. [ 3 7 11]]

  12.  

numpy.rollaxis

該函式向後滾動特定的軸,直到一個特定位置。這個函式接受三個引數:

 
  1.  
  2. numpy.rollaxis(arr, axis, start)

  3.  

其中:

  • arr:輸入陣列
  • axis:要向後滾動的軸,其它軸的相對位置不會改變
  • start:預設為零,表示完整的滾動。會滾動到特定位置。

例子

 
  1.  
  2. # 建立了三維的 ndarray

  3. import numpy as np

  4. a = np.arange(8).reshape(2,2,2)

  5.  
  6. print '原陣列:'

  7. print a

  8. print '\n'

  9. # 將軸 2 滾動到軸 0(寬度到深度)

  10.  
  11. print '呼叫 rollaxis 函式:'

  12. print np.rollaxis(a,2)

  13. # 將軸 0 滾動到軸 1:(寬度到高度)

  14. print '\n'

  15.  
  16. print '呼叫 rollaxis 函式:'

  17. print np.rollaxis(a,2,1)

  18.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [[[0 1]

  4. [2 3]]

  5. [[4 5]

  6. [6 7]]]

  7.  
  8. 呼叫 rollaxis 函式:

  9. [[[0 2]

  10. [4 6]]

  11. [[1 3]

  12. [5 7]]]

  13.  
  14. 呼叫 rollaxis 函式:

  15. [[[0 2]

  16. [1 3]]

  17. [[4 6]

  18. [5 7]]]

  19.  

numpy.swapaxes

該函式交換陣列的兩個軸。對於 1.10 之前的 NumPy 版本,會返回交換後陣列的試圖。這個函式接受下列引數:

 
  1.  
  2. numpy.swapaxes(arr, axis1, axis2)

  3.  
  • arr:要交換其軸的輸入陣列
  • axis1:對應第一個軸的整數
  • axis2:對應第二個軸的整數
 
  1.  
  2. # 建立了三維的 ndarray

  3. import numpy as np

  4. a = np.arange(8).reshape(2,2,2)

  5.  
  6. print '原陣列:'

  7. print a

  8. print '\n'

  9. # 現在交換軸 0(深度方向)到軸 2(寬度方向)

  10.  
  11. print '呼叫 swapaxes 函式後的陣列:'

  12. print np.swapaxes(a, 2, 0)

  13.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [[[0 1]

  4. [2 3]]

  5.  
  6. [[4 5]

  7. [6 7]]]

  8.  
  9. 呼叫 swapaxes 函式後的陣列:

  10. [[[0 4]

  11. [2 6]]

  12.  
  13. [[1 5]

  14. [3 7]]]

  15.  

修改維度

序號維度和描述
1.broadcast 產生模仿廣播的物件
2.broadcast_to 將陣列廣播到新形狀
3.expand_dims 擴充套件陣列的形狀
4.squeeze 從陣列的形狀中刪除單維條目

broadcast

如前所述,NumPy 已經內建了對廣播的支援。 此功能模仿廣播機制。 它返回一個物件,該物件封裝了將一個陣列廣播到另一個陣列的結果。

該函式使用兩個陣列作為輸入引數。 下面的例子說明了它的用法。

 
  1.  
  2. import numpy as np

  3. x = np.array([[1], [2], [3]])

  4. y = np.array([4, 5, 6])

  5.  
  6. # 對 y 廣播 x

  7. b = np.broadcast(x,y)

  8. # 它擁有 iterator 屬性,基於自身元件的迭代器元組

  9.  
  10. print '對 y 廣播 x:'

  11. r,c = b.iters

  12. print r.next(), c.next()

  13. print r.next(), c.next()

  14. print '\n'

  15. # shape 屬性返回廣播物件的形狀

  16.  
  17. print '廣播物件的形狀:'

  18. print b.shape

  19. print '\n'

  20. # 手動使用 broadcast 將 x 與 y 相加

  21. b = np.broadcast(x,y)

  22. c = np.empty(b.shape)

  23.  
  24. print '手動使用 broadcast 將 x 與 y 相加:'

  25. print c.shape

  26. print '\n'

  27. c.flat = [u + v for (u,v) in b]

  28.  
  29. print '呼叫 flat 函式:'

  30. print c

  31. print '\n'

  32. # 獲得了和 NumPy 內建的廣播支援相同的結果

  33.  
  34. print 'x 與 y 的和:'

  35. print x + y

  36.  

輸出如下:

 
  1.  
  2. 對 y 廣播 x:

  3. 1 4

  4. 1 5

  5.  
  6. 廣播物件的形狀:

  7. (3, 3)

  8.  
  9. 手動使用 broadcast 將 x 與 y 相加:

  10. (3, 3)

  11.  
  12. 呼叫 flat 函式:

  13. [[ 5. 6. 7.]

  14. [ 6. 7. 8.]

  15. [ 7. 8. 9.]]

  16.  
  17. x 與 y 的和:

  18. [[5 6 7]

  19. [6 7 8]

  20. [7 8 9]]

  21.  

numpy.broadcast_to

此函式將陣列廣播到新形狀。 它在原始陣列上返回只讀檢視。 它通常不連續。 如果新形狀不符合 NumPy 的廣播規則,該函式可能會丟擲ValueError

注意 - 此功能可用於 1.10.0 及以後的版本。

該函式接受以下引數。

 
  1.  
  2. numpy.broadcast_to(array, shape, subok)

  3.  

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(4).reshape(1,4)

  4.  
  5. print '原陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '呼叫 broadcast_to 函式之後:'

  10. print np.broadcast_to(a,(4,4))

  11.  

輸出如下:

 
  1. [[0 1 2 3]

  2. [0 1 2 3]

  3. [0 1 2 3]

  4. [0 1 2 3]]

numpy.expand_dims

函式通過在指定位置插入新的軸來擴充套件陣列形狀。該函式需要兩個引數:

 
  1.  
  2. numpy.expand_dims(arr, axis)

  3.  

其中:

  • arr:輸入陣列
  • axis:新軸插入的位置

例子

 
  1.  
  2. import numpy as np

  3. x = np.array(([1,2],[3,4]))

  4.  
  5. print '陣列 x:'

  6. print x

  7. print '\n'

  8. y = np.expand_dims(x, axis = 0)

  9.  
  10. print '陣列 y:'

  11. print y

  12. print '\n'

  13.  
  14. print '陣列 x 和 y 的形狀:'

  15. print x.shape, y.shape

  16. print '\n'

  17. # 在位置 1 插入軸

  18. y = np.expand_dims(x, axis = 1)

  19.  
  20. print '在位置 1 插入軸之後的陣列 y:'

  21. print y

  22. print '\n'

  23.  
  24. print 'x.ndim 和 y.ndim:'

  25. print x.ndim,y.ndim

  26. print '\n'

  27.  
  28. print 'x.shape 和 y.shape:'

  29. print x.shape, y.shape

  30.  

輸出如下:

 
  1.  
  2. 陣列 x:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 陣列 y:

  7. [[[1 2]

  8. [3 4]]]

  9.  
  10. 陣列 x 和 y 的形狀:

  11. (2, 2) (1, 2, 2)

  12.  
  13. 在位置 1 插入軸之後的陣列 y:

  14. [[[1 2]]

  15. [[3 4]]]

  16.  
  17. x.shape 和 y.shape:

  18. 2 3

  19.  
  20. x.shape and y.shape:

  21. (2, 2) (2, 1, 2)

  22.  

numpy.squeeze

函式從給定陣列的形狀中刪除一維條目。 此函式需要兩個引數。

 
  1.  
  2. numpy.squeeze(arr, axis)

  3.  

其中:

  • arr:輸入陣列
  • axis:整數或整數元組,用於選擇形狀中單一維度條目的子集

例子

 
  1.  
  2. import numpy as np

  3. x = np.arange(9).reshape(1,3,3)

  4.  
  5. print '陣列 x:'

  6. print x

  7. print '\n'

  8. y = np.squeeze(x)

  9.  
  10. print '陣列 y:'

  11. print y

  12. print '\n'

  13.  
  14. print '陣列 x 和 y 的形狀:'

  15. print x.shape, y.shape

  16.  

輸出如下:

 
  1.  
  2. 陣列 x:

  3. [[[0 1 2]

  4. [3 4 5]

  5. [6 7 8]]]

  6.  
  7. 陣列 y:

  8. [[0 1 2]

  9. [3 4 5]

  10. [6 7 8]]

  11.  
  12. 陣列 x 和 y 的形狀:

  13. (1, 3, 3) (3, 3)

  14.  

陣列的連線

序號陣列及描述
1.concatenate 沿著現存的軸連線資料序列
2.stack 沿著新軸連線陣列序列
3.hstack 水平堆疊序列中的陣列(列方向)
4.vstack 豎直堆疊序列中的陣列(行方向)

numpy.concatenate

陣列的連線是指連線。 此函式用於沿指定軸連線相同形狀的兩個或多個陣列。 該函式接受以下引數。

 
  1.  
  2. numpy.concatenate((a1, a2, ...), axis)

  3.  

其中:

  • a1, a2, ...:相同型別的陣列序列
  • axis:沿著它連線陣列的軸,預設為 0

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. b = np.array([[5,6],[7,8]])

  9.  
  10. print '第二個陣列:'

  11. print b

  12. print '\n'

  13. # 兩個陣列的維度相同

  14.  
  15. print '沿軸 0 連線兩個陣列:'

  16. print np.concatenate((a,b))

  17. print '\n'

  18.  
  19. print '沿軸 1 連線兩個陣列:'

  20. print np.concatenate((a,b),axis = 1)

  21.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 第二個陣列:

  7. [[5 6]

  8. [7 8]]

  9.  
  10. 沿軸 0 連線兩個陣列:

  11. [[1 2]

  12. [3 4]

  13. [5 6]

  14. [7 8]]

  15.  
  16. 沿軸 1 連線兩個陣列:

  17. [[1 2 5 6]

  18. [3 4 7 8]]

  19.  

numpy.stack

此函式沿新軸連線陣列序列。 此功能新增自 NumPy 版本 1.10.0。 需要提供以下引數。

 
  1.  
  2. numpy.stack(arrays, axis)

  3.  

其中:

  • arrays:相同形狀的陣列序列
  • axis:返回陣列中的軸,輸入陣列沿著它來堆疊
 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. b = np.array([[5,6],[7,8]])

  9.  
  10. print '第二個陣列:'

  11. print b

  12. print '\n'

  13.  
  14. print '沿軸 0 堆疊兩個陣列:'

  15. print np.stack((a,b),0)

  16. print '\n'

  17.  
  18. print '沿軸 1 堆疊兩個陣列:'

  19. print np.stack((a,b),1)

  20.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 第二個陣列:

  7. [[5 6]

  8. [7 8]]

  9.  
  10. 沿軸 0 堆疊兩個陣列:

  11. [[[1 2]

  12. [3 4]]

  13. [[5 6]

  14. [7 8]]]

  15.  
  16. 沿軸 1 堆疊兩個陣列:

  17. [[[1 2]

  18. [5 6]]

  19. [[3 4]

  20. [7 8]]]

  21.  

numpy.hstack

numpy.stack函式的變體,通過堆疊來生成水平的單個陣列。

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. b = np.array([[5,6],[7,8]])

  9.  
  10. print '第二個陣列:'

  11. print b

  12. print '\n'

  13.  
  14. print '水平堆疊:'

  15. c = np.hstack((a,b))

  16. print c

  17. print '\n'

  18.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 第二個陣列:

  7. [[5 6]

  8. [7 8]]

  9.  
  10. 水平堆疊:

  11. [[1 2 5 6]

  12. [3 4 7 8]]

  13.  

numpy.vstack

numpy.stack函式的變體,通過堆疊來生成豎直的單個陣列。

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. b = np.array([[5,6],[7,8]])

  9.  
  10. print '第二個陣列:'

  11. print b

  12. print '\n'

  13.  
  14. print '豎直堆疊:'

  15. c = np.vstack((a,b))

  16. print c

  17.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 第二個陣列:

  7. [[5 6]

  8. [7 8]]

  9.  
  10. 豎直堆疊:

  11. [[1 2]

  12. [3 4]

  13. [5 6]

  14. [7 8]]

  15.  

陣列分割

序號陣列及操作
1.split 將一個陣列分割為多個子陣列
2.hsplit 將一個陣列水平分割為多個子陣列(按列)
3.vsplit 將一個陣列豎直分割為多個子陣列(按行)

numpy.split

該函式沿特定的軸將陣列分割為子陣列。函式接受三個引數:

 
  1.  
  2. numpy.split(ary, indices_or_sections, axis)

  3.  

其中:

  • ary:被分割的輸入陣列
  • indices_or_sections:可以是整數,表明要從輸入陣列建立的,等大小的子陣列的數量。 如果此引數是一維陣列,則其元素表明要建立新子陣列的點。
  • axis:預設為 0

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(9)

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '將陣列分為三個大小相等的子陣列:'

  10. b = np.split(a,3)

  11. print b

  12. print '\n'

  13.  
  14. print '將陣列在一維陣列中表明的位置分割:'

  15. b = np.split(a,[4,7])

  16. print b

  17.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [0 1 2 3 4 5 6 7 8]

  4.  
  5. 將陣列分為三個大小相等的子陣列:

  6. [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]

  7.  
  8. 將陣列在一維陣列中表明的位置分割:

  9. [array([0, 1, 2, 3]), array([4, 5, 6]), array([7, 8])]

  10.  

numpy.hsplit

numpy.hsplitsplit()函式的特例,其中軸為 1 表示水平分割,無論輸入陣列的維度是什麼。

 
  1.  
  2. import numpy as np

  3. a = np.arange(16).reshape(4,4)

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '水平分割:'

  10. b = np.hsplit(a,2)

  11. print b

  12. print '\n'

  13.  

輸出:

 
  1.  
  2. 第一個陣列:

  3. [[ 0 1 2 3]

  4. [ 4 5 6 7]

  5. [ 8 9 10 11]

  6. [12 13 14 15]]

  7.  
  8. 水平分割:

  9. [array([[ 0, 1],

  10. [ 4, 5],

  11. [ 8, 9],

  12. [12, 13]]), array([[ 2, 3],

  13. [ 6, 7],

  14. [10, 11],

  15. [14, 15]])]

  16.  

numpy.vsplit

numpy.vsplitsplit()函式的特例,其中軸為 0 表示豎直分割,無論輸入陣列的維度是什麼。下面的例子使之更清楚。

 
  1.  
  2. import numpy as np

  3. a = np.arange(16).reshape(4,4)

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '豎直分割:'

  10. b = np.vsplit(a,2)

  11. print b

  12.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[ 0 1 2 3]

  4. [ 4 5 6 7]

  5. [ 8 9 10 11]

  6. [12 13 14 15]]

  7.  
  8. 豎直分割:

  9. [array([[0, 1, 2, 3],

  10. [4, 5, 6, 7]]), array([[ 8, 9, 10, 11],

  11. [12, 13, 14, 15]])]

  12.  

新增/刪除元素

序號元素及描述
1.resize 返回指定形狀的新陣列
2.append 將值新增到陣列末尾
3.insert 沿指定軸將值插入到指定下標之前
4.delete 返回刪掉某個軸的子陣列的新陣列
5.unique 尋找陣列內的唯一元素

numpy.resize

此函式返回指定大小的新陣列。 如果新大小大於原始大小,則包含原始陣列中的元素的重複副本。 該函式接受以下引數。

 
  1.  
  2. numpy.resize(arr, shape)

  3.  

其中:

  • arr:要修改大小的輸入陣列
  • shape:返回陣列的新形狀

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[4,5,6]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '第一個陣列的形狀:'

  10. print a.shape

  11. print '\n'

  12. b = np.resize(a, (3,2))

  13.  
  14. print '第二個陣列:'

  15. print b

  16. print '\n'

  17.  
  18. print '第二個陣列的形狀:'

  19. print b.shape

  20. print '\n'

  21. # 要注意 a 的第一行在 b 中重複出現,因為尺寸變大了

  22.  
  23. print '修改第二個陣列的大小:'

  24. b = np.resize(a,(3,3))

  25. print b

  26.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2 3]

  4. [4 5 6]]

  5.  
  6. 第一個陣列的形狀:

  7. (2, 3)

  8.  
  9. 第二個陣列:

  10. [[1 2]

  11. [3 4]

  12. [5 6]]

  13.  
  14. 第二個陣列的形狀:

  15. (3, 2)

  16.  
  17. 修改第二個陣列的大小:

  18. [[1 2 3]

  19. [4 5 6]

  20. [1 2 3]]

  21.  

numpy.append

此函式在輸入陣列的末尾新增值。 附加操作不是原地的,而是分配新的陣列。 此外,輸入陣列的維度必須匹配否則將生成ValueError

函式接受下列函式:

 
  1.  
  2. numpy.append(arr, values, axis)

  3.  

其中:

  • arr:輸入陣列
  • values:要向arr新增的值,比如和arr形狀相同(除了要新增的軸)
  • axis:沿著它完成操作的軸。如果沒有提供,兩個引數都會被展開。

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[4,5,6]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '向陣列新增元素:'

  10. print np.append(a, [7,8,9])

  11. print '\n'

  12.  
  13. print '沿軸 0 新增元素:'

  14. print np.append(a, [[7,8,9]],axis = 0)

  15. print '\n'

  16.  
  17. print '沿軸 1 新增元素:'

  18. print np.append(a, [[5,5,5],[7,8,9]],axis = 1)

  19.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[1 2 3]

  4. [4 5 6]]

  5.  
  6. 向陣列新增元素:

  7. [1 2 3 4 5 6 7 8 9]

  8.  
  9. 沿軸 0 新增元素:

  10. [[1 2 3]

  11. [4 5 6]

  12. [7 8 9]]

  13.  
  14. 沿軸 1 新增元素:

  15. [[1 2 3 5 5 5]

  16. [4 5 6 7 8 9]]

  17.  

numpy.insert

此函式在給定索引之前,沿給定軸在輸入陣列中插入值。 如果值的型別轉換為要插入,則它與輸入陣列不同。 插入沒有原地的,函式會返回一個新陣列。 此外,如果未提供軸,則輸入陣列會被展開。

insert()函式接受以下引數:

 
  1.  
  2. numpy.insert(arr, obj, values, axis)

  3.  

其中:

  • arr:輸入陣列
  • obj:在其之前插入值的索引
  • values:要插入的值
  • axis:沿著它插入的軸,如果未提供,則輸入陣列會被展開

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4],[5,6]])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '未傳遞 Axis 引數。 在插入之前輸入陣列會被展開。'

  10. print np.insert(a,3,[11,12])

  11. print '\n'

  12. print '傳遞了 Axis 引數。 會廣播值陣列來配輸入陣列。'

  13.  
  14. print '沿軸 0 廣播:'

  15. print np.insert(a,1,[11],axis = 0)

  16. print '\n'

  17.  
  18. print '沿軸 1 廣播:'

  19. print np.insert(a,1,11,axis = 1)

  20.  

numpy.delete

此函式返回從輸入陣列中刪除指定子陣列的新陣列。 與insert()函式的情況一樣,如果未提供軸引數,則輸入陣列將展開。 該函式接受以下引數:

 
  1.  
  2. Numpy.delete(arr, obj, axis)

  3.  

其中:

  • arr:輸入陣列
  • obj:可以被切片,整數或者整數陣列,表明要從輸入陣列刪除的子陣列
  • axis:沿著它刪除給定子陣列的軸,如果未提供,則輸入陣列會被展開

例子

 
  1.  
  2. import numpy as np

  3. a = np.arange(12).reshape(3,4)

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '未傳遞 Axis 引數。 在插入之前輸入陣列會被展開。'

  10. print np.delete(a,5)

  11. print '\n'

  12.  
  13. print '刪除第二列:'

  14. print np.delete(a,1,axis = 1)

  15. print '\n'

  16.  
  17. print '包含從陣列中刪除的替代值的切片:'

  18. a = np.array([1,2,3,4,5,6,7,8,9,10])

  19. print np.delete(a, np.s_[::2])

  20.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[ 0 1 2 3]

  4. [ 4 5 6 7]

  5. [ 8 9 10 11]]

  6.  
  7. 未傳遞 Axis 引數。 在插入之前輸入陣列會被展開。

  8. [ 0 1 2 3 4 6 7 8 9 10 11]

  9.  
  10. 刪除第二列:

  11. [[ 0 2 3]

  12. [ 4 6 7]

  13. [ 8 10 11]]

  14.  
  15. 包含從陣列中刪除的替代值的切片:

  16. [ 2 4 6 8 10]

  17.  

numpy.unique

此函式返回輸入陣列中的去重元素陣列。 該函式能夠返回一個元組,包含去重陣列和相關索引的陣列。 索引的性質取決於函式呼叫中返回引數的型別。

 
  1.  
  2. numpy.unique(arr, return_index, return_inverse, return_counts)

  3.  

其中:

  • arr:輸入陣列,如果不是一維陣列則會展開
  • return_index:如果為true,返回輸入陣列中的元素下標
  • return_inverse:如果為true,返回去重陣列的下標,它可以用於重構輸入陣列
  • return_counts:如果為true,返回去重陣列中的元素在原陣列中的出現次數

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([5,2,6,2,7,5,6,8,2,9])

  4.  
  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8.  
  9. print '第一個陣列的去重值:'

  10. u = np.unique(a)

  11. print u

  12. print '\n'

  13.  
  14. print '去重陣列的索引陣列:'

  15. u,indices = np.unique(a, return_index = True)

  16. print indices

  17. print '\n'

  18.  
  19. print '我們可以看到每個和原陣列下標對應的數值:'

  20. print a

  21. print '\n'

  22.  
  23. print '去重陣列的下標:'

  24. u,indices = np.unique(a,return_inverse = True)

  25. print u

  26. print '\n'

  27.  
  28. print '下標為:'

  29. print indices

  30. print '\n'

  31.  
  32. print '使用下標重構原陣列:'

  33. print u[indices]

  34. print '\n'

  35.  
  36. print '返回去重元素的重複數量:'

  37. u,indices = np.unique(a,return_counts = True)

  38. print u

  39. print indices

  40.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [5 2 6 2 7 5 6 8 2 9]

  4.  
  5. 第一個陣列的去重值:

  6. [2 5 6 7 8 9]

  7.  
  8. 去重陣列的索引陣列:

  9. [1 0 2 4 7 9]

  10.  
  11. 我們可以看到每個和原陣列下標對應的數值:

  12. [5 2 6 2 7 5 6 8 2 9]

  13.  
  14. 去重陣列的下標:

  15. [2 5 6 7 8 9]

  16.  
  17. 下標為:

  18. [1 0 2 0 3 1 2 4 0 5]

  19.  
  20. 使用下標重構原陣列:

  21. [5 2 6 2 7 5 6 8 2 9]

  22.  
  23. 返回唯一元素的重複數量:

  24. [2 5 6 7 8 9]

  25. [3 2 2 1 1 1]

  26.  

NumPy - 位操作

下面是 NumPy 包中可用的位操作函式。

序號操作及描述
1.bitwise_and 對陣列元素執行位與操作
2.bitwise_or 對陣列元素執行位或操作
3.invert 計算位非
4.left_shift 向左移動二進位制表示的位
5.right_shift 向右移動二進位制表示的位

bitwise_and

通過np.bitwise_and()函式對輸入陣列中的整數的二進位制表示的相應位執行位與運算。

例子

 
  1.  
  2. import numpy as np

  3. print '13 和 17 的二進位制形式:'

  4. a,b = 13,17

  5. print bin(a), bin(b)

  6. print '\n'

  7.  
  8. print '13 和 17 的位與:'

  9. print np.bitwise_and(13, 17)

  10.  

輸出如下:

 
  1. 13 和 17 的二進位制形式:

  2. 0b1101 0b10001

  3.  
  4. 13 和 17 的位與:

  5. 1

你可以使用下表驗證此輸出。 考慮下面的位與真值表。

ABAND
111
100
010
000

| | | 1 | 1 | 0 | 1 |
| --- | --- |
| AND |
| | 1 | 0 | 0 | 0 | 1 |
| result | 0 | 0 | 0 | 0 | 1 |

bitwise_or

通過np.bitwise_or()函式對輸入陣列中的整數的二進位制表示的相應位執行位或運算。

例子

 
  1.  
  2. import numpy as np

  3. a,b = 13,17

  4. print '13 和 17 的二進位制形式:'

  5. print bin(a), bin(b)

  6.  
  7. print '13 和 17 的位或:'

  8. print np.bitwise_or(13, 17)

  9.  

輸出如下:

 
  1. 13 和 17 的二進位制形式:

  2. 0b1101 0b10001

  3.  
  4. 13 和 17 的位或:

  5. 29

你可以使用下表驗證此輸出。 考慮下面的位或真值表。

ABOR
111
101
011
000

| | | 1 | 1 | 0 | 1 |
| --- | --- |
| OR |
| | 1 | 0 | 0 | 0 | 1 |
| result | 1 | 1 | 1 | 0 | 1 |

invert

此函式計算輸入陣列中整數的位非結果。 對於有符號整數,返回補碼。

例子

 
  1.  
  2. import numpy as np

  3.  
  4. print '13 的位反轉,其中 ndarray 的 dtype 是 uint8:'

  5. print np.invert(np.array([13], dtype = np.uint8))

  6. print '\n'

  7. # 比較 13 和 242 的二進位制表示,我們發現了位的反轉

  8.  
  9. print '13 的二進位制表示:'

  10. print np.binary_repr(13, width = 8)

  11. print '\n'

  12.  
  13. print '242 的二進位制表示:'

  14. print np.binary_repr(242, width = 8)

  15.  

輸出如下:

 
  1.  
  2. 13 的位反轉,其中 ndarray 的 dtype 是 uint8:

  3. [242]

  4.  
  5. 13 的二進位制表示:

  6. 00001101

  7.  
  8. 242 的二進位制表示:

  9. 11110010

  10.  

請注意,np.binary_repr()函式返回給定寬度中十進位制數的二進位制表示。

left_shift

numpy.left shift()函式將陣列元素的二進位制表示中的位向左移動到指定位置,右側附加相等數量的 0。

例如,

 
  1.  
  2. import numpy as np

  3.  
  4. print '將 10 左移兩位:'

  5. print np.left_shift(10,2)

  6. print '\n'

  7.  
  8. print '10 的二進位制表示:'

  9. print np.binary_repr(10, width = 8)

  10. print '\n'

  11.  
  12. print '40 的二進位制表示:'

  13. print np.binary_repr(40, width = 8)

  14. # '00001010' 中的兩位移動到了左邊,並在右邊新增了兩個 0。

  15.  

輸出如下:

 
  1. 將 10 左移兩位:

  2. 40

  3.  
  4. 10 的二進位制表示:

  5. 00001010

  6.  
  7. 40 的二進位制表示:

  8. 00101000

right_shift

numpy.right_shift()函式將陣列元素的二進位制表示中的位向右移動到指定位置,左側附加相等數量的 0。

 
  1.  
  2. import numpy as np

  3.  
  4. print '將 40 右移兩位:'

  5. print np.right_shift(40,2)

  6. print '\n'

  7.  
  8. print '40 的二進位制表示:'

  9. print np.binary_repr(40, width = 8)

  10. print '\n'

  11.  
  12. print '10 的二進位制表示:'

  13. print np.binary_repr(10, width = 8)

  14. # '00001010' 中的兩位移動到了右邊,並在左邊新增了兩個 0。

  15.  

輸出如下:

 
  1. 將 40 右移兩位:

  2. 10

  3.  
  4. 40 的二進位制表示:

  5. 00101000

  6.  
  7. 10 的二進位制表示:

  8. 00001010

NumPy - 字串函式

以下函式用於對dtypenumpy.string_numpy.unicode_的陣列執行向量化字串操作。 它們基於 Python 內建庫中的標準字串函式。

序號函式及描述
1.add() 返回兩個strUnicode陣列的逐個字串連線
2.multiply() 返回按元素多重連線後的字串
3.center() 返回給定字串的副本,其中元素位於特定字串的中央
4.capitalize() 返回給定字串的副本,其中只有第一個字串大寫
5.title() 返回字串或 Unicode 的按元素標題轉換版本
6.lower() 返回一個陣列,其元素轉換為小寫
7.upper() 返回一個陣列,其元素轉換為大寫
8.split() 返回字串中的單詞列表,並使用分隔符來分割
9.splitlines() 返回元素中的行列表,以換行符分割
10.strip() 返回陣列副本,其中元素移除了開頭或者結尾處的特定字元
11.join() 返回一個字串,它是序列中字串的連線
12.replace() 返回字串的副本,其中所有子字串的出現位置都被新字串取代
13.decode() 按元素呼叫str.decode
14.encode() 按元素呼叫str.encode

這些函式在字元陣列類(numpy.char)中定義。 較舊的 Numarray 包包含chararray類。 numpy.char類中的上述函式在執行向量化字串操作時非常有用。

numpy.char.add()

函式執行按元素的字串連線。

 
  1.  
  2. import numpy as np

  3. print '連線兩個字串:'

  4. print np.char.add(['hello'],[' xyz'])

  5. print '\n'

  6.  
  7. print '連線示例:'

  8. print np.char.add(['hello', 'hi'],[' abc', ' xyz'])

  9.  

輸出如下:

 
  1. 連線兩個字串:

  2. ['hello xyz']

  3.  
  4. 連線示例:

  5. ['hello abc' 'hi xyz']

numpy.char.multiply()

這個函式執行多重連線。

 
  1.  
  2. import numpy as np

  3. print np.char.multiply('Hello ',3)

  4.  

輸出如下:

Hello Hello Hello 

numpy.char.center()

此函式返回所需寬度的陣列,以便輸入字串位於中心,並使用fillchar在左側和右側進行填充。

 
  1.  
  2. import numpy as np

  3. # np.char.center(arr, width,fillchar)

  4. print np.char.center('hello', 20,fillchar = '*')

  5.  

輸出如下:

 
  1. *******hello********

  2.  

numpy.char.capitalize()

函式返回字串的副本,其中第一個字母大寫

 
  1.  
  2. import numpy as np

  3. print np.char.capitalize('hello world')

  4.  

輸出如下:

Hello world 

numpy.char.title()

返回輸入字串的按元素標題轉換版本,其中每個單詞的首字母都大寫。

 
  1.  
  2. import numpy as np

  3. print np.char.title('hello how are you?')

  4.  

輸出如下:

Hello How Are You?

numpy.char.lower()

函式返回一個陣列,其元素轉換為小寫。它對每個元素呼叫str.lower

 
  1.  
  2. import numpy as np

  3. print np.char.lower(['HELLO','WORLD'])

  4. print np.char.lower('HELLO')

  5.  

輸出如下:

 
  1. ['hello' 'world']

  2. hello

numpy.char.upper()

函式返回一個陣列,其元素轉換為大寫。它對每個元素呼叫str.upper

 
  1.  
  2. import numpy as np

  3. print np.char.upper('hello')

  4. print np.char.upper(['hello','world'])

  5.  

輸出如下:

 
  1. HELLO

  2. ['HELLO' 'WORLD']

numpy.char.split()

此函式返回輸入字串中的單詞列表。 預設情況下,空格用作分隔符。 否則,指定的分隔符字元用於分割字串。

 
  1.  
  2. import numpy as np

  3. print np.char.split ('hello how are you?')

  4. print np.char.split ('TutorialsPoint,Hyderabad,Telangana', sep = ',')

  5.  

輸出如下:

 
  1. ['hello', 'how', 'are', 'you?']

  2. ['TutorialsPoint', 'Hyderabad', 'Telangana']

numpy.char.splitlines()

函式返回陣列中元素的單詞列表,以換行符分割。

 
  1.  
  2. import numpy as np

  3. print np.char.splitlines('hello\nhow are you?')

  4. print np.char.splitlines('hello\rhow are you?')

  5.  

輸出如下:

 
  1. ['hello', 'how are you?']

  2. ['hello', 'how are you?']

'\n''\r''\r\n'都會用作換行符。

numpy.char.strip()

函式返回陣列的副本,其中元素移除了開頭或結尾處的特定字元。

 
  1.  
  2. import numpy as np

  3. print np.char.strip('ashok arora','a')

  4. print np.char.strip(['arora','admin','java'],'a')

  5.  

輸出如下:

 
  1. shok aror

  2. ['ror' 'dmin' 'jav']

numpy.char.join()

這個函式返回一個字串,其中單個字元由特定的分隔符連線。

 
  1.  
  2. import numpy as np

  3. print np.char.join(':','dmy')

  4. print np.char.join([':','-'],['dmy','ymd'])

  5.  

輸出如下:

 
  1.  
  2. d:m:y

  3. ['d:m:y' 'y-m-d']

  4.  

numpy.char.replace()

這個函式返回字串副本,其中所有字元序列的出現位置都被另一個給定的字元序列取代。

 
  1.  
  2. import numpy as np

  3. print np.char.replace ('He is a good boy', 'is', 'was')

  4.  

輸出如下:

He was a good boy

numpy.char.decode()

這個函式在給定的字串中使用特定編碼呼叫str.decode()

 
  1.  
  2. import numpy as np

  3.  
  4. a = np.char.encode('hello', 'cp500')

  5. print a

  6. print np.char.decode(a,'cp500')

  7.  

輸出如下:

 
  1. \x88\x85\x93\x93\x96

  2. hello

numpy.char.encode()

此函式對陣列中的每個元素呼叫str.encode函式。 預設編碼是utf_8,可以使用標準 Python 庫中的編解碼器。

 
  1.  
  2. import numpy as np

  3. a = np.char.encode('hello', 'cp500')

  4. print a

  5.  

輸出如下:

\x88\x85\x93\x93\x96

NumPy - 算數函式

很容易理解的是,NumPy 包含大量的各種數學運算功能。 NumPy 提供標準的三角函式,算術運算的函式,複數處理函式等。

三角函式

NumPy 擁有標準的三角函式,它為弧度制單位的給定角度返回三角函式比值。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([0,30,45,60,90])

  4. print '不同角度的正弦值:'

  5. # 通過乘 pi/180 轉化為弧度

  6. print np.sin(a*np.pi/180)

  7. print '\n'

  8. print '陣列中角度的餘弦值:'

  9. print np.cos(a*np.pi/180)

  10. print '\n'

  11. print '陣列中角度的正切值:'

  12. print np.tan(a*np.pi/180)

  13.  

輸出如下:

 
  1. 不同角度的正弦值:

  2. [ 0. 0.5 0.70710678 0.8660254 1. ]

  3.  
  4. 陣列中角度的餘弦值:

  5. [ 1.00000000e+00 8.66025404e-01 7.07106781e-01 5.00000000e-01

  6. 6.12323400e-17]

  7.  
  8. 陣列中角度的正切值:

  9. [ 0.00000000e+00 5.77350269e-01 1.00000000e+00 1.73205081e+00

  10. 1.63312394e+16]

  11.  

arcsinarccos,和arctan函式返回給定角度的sincostan的反三角函式。 這些函式的結果可以通過numpy.degrees()函式通過將弧度制轉換為角度制來驗證。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([0,30,45,60,90])

  4. print '含有正弦值的陣列:'

  5. sin = np.sin(a*np.pi/180)

  6. print sin

  7. print '\n'

  8. print '計算角度的反正弦,返回值以弧度為單位:'

  9. inv = np.arcsin(sin)

  10. print inv

  11. print '\n'

  12. print '通過轉化為角度制來檢查結果:'

  13. print np.degrees(inv)

  14. print '\n'

  15. print 'arccos 和 arctan 函式行為類似:'

  16. cos = np.cos(a*np.pi/180)

  17. print cos

  18. print '\n'

  19. print '反餘弦:'

  20. inv = np.arccos(cos)

  21. print inv

  22. print '\n'

  23. print '角度制單位:'

  24. print np.degrees(inv)

  25. print '\n'

  26. print 'tan 函式:'

  27. tan = np.tan(a*np.pi/180)

  28. print tan

  29. print '\n'

  30. print '反正切:'

  31. inv = np.arctan(tan)

  32. print inv

  33. print '\n'

  34. print '角度制單位:'

  35. print np.degrees(inv)

  36.  

輸出如下:

 
  1.  
  2. 含有正弦值的陣列:

  3. [ 0. 0.5 0.70710678 0.8660254 1. ]

  4.  
  5. 計算角度的反正弦,返回值以弧度製為單位:

  6. [ 0. 0.52359878 0.78539816 1.04719755 1.57079633]

  7.  
  8. 通過轉化為角度制來檢查結果:

  9. [ 0. 30. 45. 60. 90.]

  10.  
  11. arccos 和 arctan 函式行為類似:

  12. [ 1.00000000e+00 8.66025404e-01 7.07106781e-01 5.00000000e-01

  13. 6.12323400e-17]

  14.  
  15. 反餘弦:

  16. [ 0. 0.52359878 0.78539816 1.04719755 1.57079633]

  17.  
  18. 角度制單位:

  19. [ 0. 30. 45. 60. 90.]

  20.  
  21. tan 函式:

  22. [ 0.00000000e+00 5.77350269e-01 1.00000000e+00 1.73205081e+00

  23. 1.63312394e+16]

  24.  
  25. 反正切:

  26. [ 0. 0.52359878 0.78539816 1.04719755 1.57079633]

  27.  
  28. 角度制單位:

  29. [ 0. 30. 45. 60. 90.]

  30.  
  31.  

舍入函式

numpy.around()

這個函式返回四捨五入到所需精度的值。 該函式接受以下引數。

 
  1.  
  2. numpy.around(a,decimals)

  3.  
  4.  

其中:

序號引數及描述
1.a 輸入陣列
2.decimals 要舍入的小數位數。 預設值為0。 如果為負,整數將四捨五入到小數點左側的位置

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([1.0,5.55, 123, 0.567, 25.532])

  4. print '原陣列:'

  5. print a

  6. print '\n'

  7. print '舍入後:'

  8. print np.around(a)

  9. print np.around(a, decimals = 1)

  10. print np.around(a, decimals = -1)

  11.  

輸出如下:

 
  1.  
  2. 原陣列:

  3. [ 1. 5.55 123. 0.567 25.532]

  4.  
  5. 舍入後:

  6. [ 1. 6. 123. 1. 26. ]

  7. [ 1. 5.6 123. 0.6 25.5]

  8. [ 0. 10. 120. 0. 30. ]

  9.  
  10.  

numpy.floor()

此函式返回不大於輸入引數的最大整數。 即標量x 的下限是最大的整數i ,使得i <= x。 注意在Python中,向下取整總是從 0 舍入。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([-1.7, 1.5, -0.2, 0.6, 10])

  4. print '提供的陣列:'

  5. print a

  6. print '\n'

  7. print '修改後的陣列:'

  8. print np.floor(a)

  9.  

輸出如下:

 
  1.  
  2. 提供的陣列:

  3. [ -1.7 1.5 -0.2 0.6 10. ]

  4.  
  5. 修改後的陣列:

  6. [ -2. 1. -1. 0. 10.]

  7.  
  8.  

numpy.ceil()

ceil()函式返回輸入值的上限,即,標量x的上限是最小的整數i ,使得i> = x

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([-1.7, 1.5, -0.2, 0.6, 10])

  4. print '提供的陣列:'

  5. print a

  6. print '\n'

  7. print '修改後的陣列:'

  8. print np.ceil(a)

  9.  

輸出如下:

 
  1.  
  2. 提供的陣列:

  3. [ -1.7 1.5 -0.2 0.6 10. ]

  4.  
  5. 修改後的陣列:

  6. [ -1. 2. -0. 1. 10.]

  7.  
  8.  

NumPy - 算數運算

用於執行算術運算(如add()subtract()multiply()divide())的輸入陣列必須具有相同的形狀或符合陣列廣播規則。

示例

 
  1.  
  2. import numpy as np

  3. a = np.arange(9, dtype = np.float_).reshape(3,3)

  4. print '第一個陣列:'

  5. print a

  6. print '\n'

  7. print '第二個陣列:'

  8. b = np.array([10,10,10])

  9. print b

  10. print '\n'

  11. print '兩個陣列相加:'

  12. print np.add(a,b)

  13. print '\n'

  14. print '兩個陣列相減:'

  15. print np.subtract(a,b)

  16. print '\n'

  17. print '兩個陣列相乘:'

  18. print np.multiply(a,b)

  19. print '\n'

  20. print '兩個陣列相除:'

  21. print np.divide(a,b)

  22.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [[ 0. 1. 2.]

  4. [ 3. 4. 5.]

  5. [ 6. 7. 8.]]

  6.  
  7. 第二個陣列:

  8. [10 10 10]

  9.  
  10. 兩個陣列相加:

  11. [[ 10. 11. 12.]

  12. [ 13. 14. 15.]

  13. [ 16. 17. 18.]]

  14.  
  15. 兩個陣列相減:

  16. [[-10. -9. -8.]

  17. [ -7. -6. -5.]

  18. [ -4. -3. -2.]]

  19.  
  20. 兩個陣列相乘:

  21. [[ 0. 10. 20.]

  22. [ 30. 40. 50.]

  23. [ 60. 70. 80.]]

  24.  
  25. 兩個陣列相除:

  26. [[ 0. 0.1 0.2]

  27. [ 0.3 0.4 0.5]

  28. [ 0.6 0.7 0.8]]

  29.  
  30.  

讓我們現在來討論 NumPy 中提供的一些其他重要的算術函式。

numpy.reciprocal()

此函式返回引數逐元素的倒數,。 由於 Python 處理整數除法的方式,對於絕對值大於 1 的整數元素,結果始終為 0, 對於整數 0,則發出溢位警告。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([0.25, 1.33, 1, 0, 100])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 reciprocal 函式:'

  8. print np.reciprocal(a)

  9. print '\n'

  10. b = np.array([100], dtype = int)

  11. print '第二個陣列:'

  12. print b

  13. print '\n'

  14. print '呼叫 reciprocal 函式:'

  15. print np.reciprocal(b)

  16.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [ 0.25 1.33 1. 0. 100. ]

  4.  
  5. 呼叫 reciprocal 函式:

  6. main.py:9: RuntimeWarning: divide by zero encountered in reciprocal

  7. print np.reciprocal(a)

  8. [ 4. 0.7518797 1. inf 0.01 ]

  9.  
  10. 第二個陣列:

  11. [100]

  12.  
  13. 呼叫 reciprocal 函式:

  14. [0]

  15.  
  16.  

numpy.power()

此函式將第一個輸入陣列中的元素作為底數,計算它與第二個輸入陣列中相應元素的冪。

 
  1.  
  2. import numpy as np

  3. a = np.array([10,100,1000])

  4. print '我們的陣列是;'

  5. print a

  6. print '\n'

  7. print '呼叫 power 函式:'

  8. print np.power(a,2)

  9. print '\n'

  10. print '第二個陣列:'

  11. b = np.array([1,2,3])

  12. print b

  13. print '\n'

  14. print '再次呼叫 power 函式:'

  15. print np.power(a,b)

  16.  

輸出如下:

 
  1.  
  2. 我們的陣列是;

  3. [ 10 100 1000]

  4.  
  5. 呼叫 power 函式:

  6. [ 100 10000 1000000]

  7.  
  8. 第二個陣列:

  9. [1 2 3]

  10.  
  11. 再次呼叫 power 函式:

  12. [ 10 10000 1000000000]

  13.  
  14.  

numpy.mod()

此函式返回輸入陣列中相應元素的除法餘數。 函式numpy.remainder()也產生相同的結果。

 
  1.  
  2. import numpy as np

  3. a = np.array([10,20,30])

  4. b = np.array([3,5,7])

  5. print '第一個陣列:'

  6. print a

  7. print '\n'

  8. print '第二個陣列:'

  9. print b

  10. print '\n'

  11. print '呼叫 mod() 函式:'

  12. print np.mod(a,b)

  13. print '\n'

  14. print '呼叫 remainder() 函式:'

  15. print np.remainder(a,b)

  16.  

輸出如下:

 
  1.  
  2. 第一個陣列:

  3. [10 20 30]

  4.  
  5. 第二個陣列:

  6. [3 5 7]

  7.  
  8. 呼叫 mod() 函式:

  9. [1 0 2]

  10.  
  11. 呼叫 remainder() 函式:

  12. [1 0 2]

  13.  
  14.  

以下函式用於對含有複數的陣列執行操作。

  • numpy.real() 返回複數型別引數的實部。

  • numpy.imag() 返回複數型別引數的虛部。

  • numpy.conj() 返回通過改變虛部的符號而獲得的共軛複數。

  • numpy.angle() 返回複數引數的角度。 函式的引數是degree。 如果為true,返回的角度以角度制來表示,否則為以弧度制來表示。

 
  1.  
  2. import numpy as np

  3. a = np.array([-5.6j, 0.2j, 11. , 1+1j])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 real() 函式:'

  8. print np.real(a)

  9. print '\n'

  10. print '呼叫 imag() 函式:'

  11. print np.imag(a)

  12. print '\n'

  13. print '呼叫 conj() 函式:'

  14. print np.conj(a)

  15. print '\n'

  16. print '呼叫 angle() 函式:'

  17. print np.angle(a)

  18. print '\n'

  19. print '再次呼叫 angle() 函式(以角度制返回):'

  20. print np.angle(a, deg = True)

  21.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [ 0.-5.6j 0.+0.2j 11.+0.j 1.+1.j ]

  4.  
  5. 呼叫 real() 函式:

  6. [ 0. 0. 11. 1.]

  7.  
  8. 呼叫 imag() 函式:

  9. [-5.6 0.2 0. 1. ]

  10.  
  11. 呼叫 conj() 函式:

  12. [ 0.+5.6j 0.-0.2j 11.-0.j 1.-1.j ]

  13.  
  14. 呼叫 angle() 函式:

  15. [-1.57079633 1.57079633 0. 0.78539816]

  16.  
  17. 再次呼叫 angle() 函式(以角度制返回):

  18. [-90. 90. 0. 45.]

  19.  
  20.  

NumPy - 統計函式

NumPy 有很多有用的統計函式,用於從陣列中給定的元素中查詢最小,最大,百分標準差和方差等。 函式說明如下:

numpy.amin() 和 numpy.amax()

這些函式從給定陣列中的元素沿指定軸返回最小值和最大值。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[3,7,5],[8,4,3],[2,4,9]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 amin() 函式:'

  8. print np.amin(a,1)

  9. print '\n'

  10. print '再次呼叫 amin() 函式:'

  11. print np.amin(a,0)

  12. print '\n'

  13. print '呼叫 amax() 函式:'

  14. print np.amax(a)

  15. print '\n'

  16. print '再次呼叫 amax() 函式:'

  17. print np.amax(a, axis = 0)

  18.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[3 7 5]

  4. [8 4 3]

  5. [2 4 9]]

  6.  
  7. 呼叫 amin() 函式:

  8. [3 3 2]

  9.  
  10. 再次呼叫 amin() 函式:

  11. [2 4 3]

  12.  
  13. 呼叫 amax() 函式:

  14. 9

  15.  
  16. 再次呼叫 amax() 函式:

  17. [8 7 9]

  18.  
  19.  

numpy.ptp()

numpy.ptp()函式返回沿軸的值的範圍(最大值 - 最小值)。

 
  1.  
  2. import numpy as np

  3. a = np.array([[3,7,5],[8,4,3],[2,4,9]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 ptp() 函式:'

  8. print np.ptp(a)

  9. print '\n'

  10. print '沿軸 1 呼叫 ptp() 函式:'

  11. print np.ptp(a, axis = 1)

  12. print '\n'

  13. print '沿軸 0 呼叫 ptp() 函式:'

  14. print np.ptp(a, axis = 0)

  15.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[3 7 5]

  4. [8 4 3]

  5. [2 4 9]]

  6.  
  7. 呼叫 ptp() 函式:

  8. 7

  9.  
  10. 沿軸 1 呼叫 ptp() 函式:

  11. [4 5 7]

  12.  
  13. 沿軸 0 呼叫 ptp() 函式:

  14. [6 3 6]

  15.  
  16.  

numpy.percentile()

百分位數是統計中使用的度量,表示小於這個值得觀察值佔某個百分比。 函式numpy.percentile()接受以下引數。

 
  1.  
  2. numpy.percentile(a, q, axis)

  3.  
  4.  

其中:

序號引數及描述
1.a 輸入陣列
2.q 要計算的百分位數,在 0 ~ 100 之間
3.axis 沿著它計算百分位數的軸

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[30,40,70],[80,20,10],[50,90,60]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 percentile() 函式:'

  8. print np.percentile(a,50)

  9. print '\n'

  10. print '沿軸 1 呼叫 percentile() 函式:'

  11. print np.percentile(a,50, axis = 1)

  12. print '\n'

  13. print '沿軸 0 呼叫 percentile() 函式:'

  14. print np.percentile(a,50, axis = 0)

  15.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[30 40 70]

  4. [80 20 10]

  5. [50 90 60]]

  6.  
  7. 呼叫 percentile() 函式:

  8. 50.0

  9.  
  10. 沿軸 1 呼叫 percentile() 函式:

  11. [ 40. 20. 60.]

  12.  
  13. 沿軸 0 呼叫 percentile() 函式:

  14. [ 50. 40. 60.]

  15.  
  16.  

numpy.median()

中值定義為將資料樣本的上半部分與下半部分分開的值。 numpy.median()函式的用法如下面的程式所示。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[30,65,70],[80,95,10],[50,90,60]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 median() 函式:'

  8. print np.median(a)

  9. print '\n'

  10. print '沿軸 0 呼叫 median() 函式:'

  11. print np.median(a, axis = 0)

  12. print '\n'

  13. print '沿軸 1 呼叫 median() 函式:'

  14. print np.median(a, axis = 1)

  15.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[30 65 70]

  4. [80 95 10]

  5. [50 90 60]]

  6.  
  7. 呼叫 median() 函式:

  8. 65.0

  9.  
  10. 沿軸 0 呼叫 median() 函式:

  11. [ 50. 90. 60.]

  12.  
  13. 沿軸 1 呼叫 median() 函式:

  14. [ 65. 80. 60.]

  15.  
  16.  

numpy.mean()

算術平均值是沿軸的元素的總和除以元素的數量。 numpy.mean()函式返回陣列中元素的算術平均值。 如果提供了軸,則沿其計算。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2,3],[3,4,5],[4,5,6]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 mean() 函式:'

  8. print np.mean(a)

  9. print '\n'

  10. print '沿軸 0 呼叫 mean() 函式:'

  11. print np.mean(a, axis = 0)

  12. print '\n'

  13. print '沿軸 1 呼叫 mean() 函式:'

  14. print np.mean(a, axis = 1)

  15.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[1 2 3]

  4. [3 4 5]

  5. [4 5 6]]

  6.  
  7. 呼叫 mean() 函式:

  8. 3.66666666667

  9.  
  10. 沿軸 0 呼叫 mean() 函式:

  11. [ 2.66666667 3.66666667 4.66666667]

  12.  
  13. 沿軸 1 呼叫 mean() 函式:

  14. [ 2. 4. 5.]

  15.  
  16.  

numpy.average()

加權平均值是由每個分量乘以反映其重要性的因子得到的平均值。 numpy.average()函式根據在另一個陣列中給出的各自的權重計算陣列中元素的加權平均值。 該函式可以接受一個軸引數。 如果沒有指定軸,則陣列會被展開。

考慮陣列[1,2,3,4]和相應的權重[4,3,2,1],通過將相應元素的乘積相加,並將和除以權重的和,來計算加權平均值。

加權平均值 = (1*4+2*3+3*2+4*1)/(4+3+2+1)

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([1,2,3,4])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 average() 函式:'

  8. print np.average(a)

  9. print '\n'

  10. # 不指定權重時相當於 mean 函式

  11. wts = np.array([4,3,2,1])

  12. print '再次呼叫 average() 函式:'

  13. print np.average(a,weights = wts)

  14. print '\n'

  15. # 如果 returned 引數設為 true,則返回權重的和

  16. print '權重的和:'

  17. print np.average([1,2,3, 4],weights = [4,3,2,1], returned = True)

  18.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [1 2 3 4]

  4.  
  5. 呼叫 average() 函式:

  6. 2.5

  7.  
  8. 再次呼叫 average() 函式:

  9. 2.0

  10.  
  11. 權重的和:

  12. (2.0, 10.0)

  13.  
  14.  

在多維陣列中,可以指定用於計算的軸。

示例

 
  1.  
  2. import numpy as np

  3. a = np.arange(6).reshape(3,2)

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '修改後的陣列:'

  8. wt = np.array([3,5])

  9. print np.average(a, axis = 1, weights = wt)

  10. print '\n'

  11. print '修改後的陣列:'

  12. print np.average(a, axis = 1, weights = wt, returned = True)

  13.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[0 1]

  4. [2 3]

  5. [4 5]]

  6.  
  7. 修改後的陣列:

  8. [ 0.625 2.625 4.625]

  9.  
  10. 修改後的陣列:

  11. (array([ 0.625, 2.625, 4.625]), array([ 8., 8., 8.]))

  12.  
  13.  

標準差

標準差是與均值的偏差的平方的平均值的平方根。 標準差公式如下:

 
  1.  
  2. std = sqrt(mean((x - x.mean())**2))

  3.  
  4.  

如果陣列是[1,2,3,4],則其平均值為2.5。 因此,差的平方是[2.25,0.25,0.25,2.25],並且其平均值的平方根除以4,即sqrt(5/4)1.1180339887498949

示例

 
  1.  
  2. import numpy as np

  3. print np.std([1,2,3,4])

  4.  

輸出如下:

 
  1.  
  2. 1.1180339887498949

  3.  
  4.  

方差

方差是偏差的平方的平均值,即mean((x - x.mean())** 2)。 換句話說,標準差是方差的平方根。

示例

 
  1.  
  2. import numpy as np

  3. print np.var([1,2,3,4])

  4.  

輸出如下:

 
  1.  
  2. 1.25

  3.  
  4.  

NumPy - 排序、搜尋和計數函式

NumPy中提供了各種排序相關功能。 這些排序函式實現不同的排序演算法,每個排序演算法的特徵在於執行速度,最壞情況效能,所需的工作空間和演算法的穩定性。 下表顯示了三種排序演算法的比較。

種類速度最壞情況工作空間穩定性
'quicksort'(快速排序)1O(n^2)0
'mergesort'(歸併排序)2O(n*log(n))~n/2
'heapsort'(堆排序)3O(n*log(n))0

numpy.sort()

sort()函式返回輸入陣列的排序副本。 它有以下引數:

 
  1.  
  2. numpy.sort(a, axis, kind, order)

  3.  
  4.  

其中:

序號引數及描述
1.a 要排序的陣列
2.axis 沿著它排序陣列的軸,如果沒有陣列會被展開,沿著最後的軸排序
3.kind 預設為'quicksort'(快速排序)
4.order 如果陣列包含欄位,則是要排序的欄位

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[3,7],[9,1]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 sort() 函式:'

  8. print np.sort(a)

  9. print '\n'

  10. print '沿軸 0 排序:'

  11. print np.sort(a, axis = 0)

  12. print '\n'

  13. # 在 sort 函式中排序欄位

  14. dt = np.dtype([('name', 'S10'),('age', int)])

  15. a = np.array([("raju",21),("anil",25),("ravi", 17), ("amar",27)], dtype = dt)

  16. print '我們的陣列是:'

  17. print a

  18. print '\n'

  19. print '按 name 排序:'

  20. print np.sort(a, order = 'name')

  21.  

輸出如下:

 
  1. 我們的陣列是:

  2. [[3 7]

  3. [9 1]]

  4.  
  5. 呼叫 sort() 函式:

  6. [[3 7]

  7. [1 9]]

  8.  
  9. 沿軸 0 排序:

  10. [[3 1]

  11. [9 7]]

  12.  
  13. 我們的陣列是:

  14. [('raju', 21) ('anil', 25) ('ravi', 17) ('amar', 27)]

  15.  
  16. 按 name 排序:

  17. [('amar', 27) ('anil', 25) ('raju', 21) ('ravi', 17)]

  18.  

numpy.argsort()

numpy.argsort()函式對輸入陣列沿給定軸執行間接排序,並使用指定排序型別返回資料的索引陣列。 這個索引陣列用於構造排序後的陣列。

示例

 
  1.  
  2. import numpy as np

  3. x = np.array([3, 1, 2])

  4. print '我們的陣列是:'

  5. print x

  6. print '\n'

  7. print '對 x 呼叫 argsort() 函式:'

  8. y = np.argsort(x)

  9. print y

  10. print '\n'

  11. print '以排序後的順序重構原陣列:'

  12. print x[y]

  13. print '\n'

  14. print '使用迴圈重構原陣列:'

  15. for i in y:

  16. print x[i],

  17.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [3 1 2]

  4.  
  5. 對 x 呼叫 argsort() 函式:

  6. [1 2 0]

  7.  
  8. 以排序後的順序重構原陣列:

  9. [1 2 3]

  10.  
  11. 使用迴圈重構原陣列:

  12. 1 2 3

  13.  
  14.  

numpy.lexsort()

函式使用鍵序列執行間接排序。 鍵可以看作是電子表格中的一列。 該函式返回一個索引陣列,使用它可以獲得排序資料。 注意,最後一個鍵恰好是 sort 的主鍵。

示例

 
  1.  
  2. import numpy as np

  3.  
  4. nm = ('raju','anil','ravi','amar')

  5. dv = ('f.y.', 's.y.', 's.y.', 'f.y.')

  6. ind = np.lexsort((dv,nm))

  7. print '呼叫 lexsort() 函式:'

  8. print ind

  9. print '\n'

  10. print '使用這個索引來獲取排序後的資料:'

  11. print [nm[i] + ", " + dv[i] for i in ind]

  12.  

輸出如下:

 
  1. 呼叫 lexsort() 函式:

  2. [3 1 0 2]

  3.  
  4. 使用這個索引來獲取排序後的資料:

  5. ['amar, f.y.', 'anil, s.y.', 'raju, f.y.', 'ravi, s.y.']

  6.  

NumPy 模組有一些用於在陣列內搜尋的函式。 提供了用於找到最大值,最小值以及滿足給定條件的元素的函式。

numpy.argmax() 和 numpy.argmin()

這兩個函式分別沿給定軸返回最大和最小元素的索引。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[30,40,70],[80,20,10],[50,90,60]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 argmax() 函式:'

  8. print np.argmax(a)

  9. print '\n'

  10. print '展開陣列:'

  11. print a.flatten()

  12. print '\n'

  13. print '沿軸 0 的最大值索引:'

  14. maxindex = np.argmax(a, axis = 0)

  15. print maxindex

  16. print '\n'

  17. print '沿軸 1 的最大值索引:'

  18. maxindex = np.argmax(a, axis = 1)

  19. print maxindex

  20. print '\n'

  21. print '呼叫 argmin() 函式:'

  22. minindex = np.argmin(a)

  23. print minindex

  24. print '\n'

  25. print '展開陣列中的最小值:'

  26. print a.flatten()[minindex]

  27. print '\n'

  28. print '沿軸 0 的最小值索引:'

  29. minindex = np.argmin(a, axis = 0)

  30. print minindex

  31. print '\n'

  32. print '沿軸 1 的最小值索引:'

  33. minindex = np.argmin(a, axis = 1)

  34. print minindex

  35.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[30 40 70]

  4. [80 20 10]

  5. [50 90 60]]

  6.  
  7. 呼叫 argmax() 函式:

  8. 7

  9.  
  10. 展開陣列:

  11. [30 40 70 80 20 10 50 90 60]

  12.  
  13. 沿軸 0 的最大值索引:

  14. [1 2 0]

  15.  
  16. 沿軸 1 的最大值索引:

  17. [2 0 1]

  18.  
  19. 呼叫 argmin() 函式:

  20. 5

  21.  
  22. 展開陣列中的最小值:

  23. 10

  24.  
  25. 沿軸 0 的最小值索引:

  26. [0 1 1]

  27.  
  28. 沿軸 1 的最小值索引:

  29. [0 2 0]

  30.  
  31.  

numpy.nonzero()

numpy.nonzero()函式返回輸入陣列中非零元素的索引。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[30,40,0],[0,20,10],[50,0,60]])

  4. print '我們的陣列是:'

  5. print a

  6. print '\n'

  7. print '呼叫 nonzero() 函式:'

  8. print np.nonzero (a)

  9.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[30 40 0]

  4. [ 0 20 10]

  5. [50 0 60]]

  6.  
  7. 呼叫 nonzero() 函式:

  8. (array([0, 0, 1, 1, 2, 2]), array([0, 1, 1, 2, 0, 2]))

  9.  
  10.  

numpy.where()

where()函式返回輸入陣列中滿足給定條件的元素的索引。

示例

 
  1.  
  2. import numpy as np

  3. x = np.arange(9.).reshape(3, 3)

  4. print '我們的陣列是:'

  5. print x

  6. print '大於 3 的元素的索引:'

  7. y = np.where(x > 3)

  8. print y

  9. print '使用這些索引來獲取滿足條件的元素:'

  10. print x[y]

  11.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[ 0. 1. 2.]

  4. [ 3. 4. 5.]

  5. [ 6. 7. 8.]]

  6.  
  7. 大於 3 的元素的索引:

  8. (array([1, 1, 2, 2, 2]), array([1, 2, 0, 1, 2]))

  9.  
  10. 使用這些索引來獲取滿足條件的元素:

  11. [ 4. 5. 6. 7. 8.]

  12.  
  13.  

numpy.extract()

extract()函式返回滿足任何條件的元素。

 
  1.  
  2. import numpy as np

  3. x = np.arange(9.).reshape(3, 3)

  4. print '我們的陣列是:'

  5. print x

  6. # 定義條件

  7. condition = np.mod(x,2) == 0

  8. print '按元素的條件值:'

  9. print condition

  10. print '使用條件提取元素:'

  11. print np.extract(condition, x)

  12.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [[ 0. 1. 2.]

  4. [ 3. 4. 5.]

  5. [ 6. 7. 8.]]

  6.  
  7. 按元素的條件值:

  8. [[ True False True]

  9. [False True False]

  10. [ True False True]]

  11.  
  12. 使用條件提取元素:

  13. [ 0. 2. 4. 6. 8.]

  14.  
  15.  

NumPy - 位元組交換

我們已經看到,儲存在計算機記憶體中的資料取決於 CPU 使用的架構。 它可以是小端(最小有效位儲存在最小地址中)或大端(最小有效位元組儲存在最大地址中)。

numpy.ndarray.byteswap()

numpy.ndarray.byteswap()函式在兩個表示:大端和小端之間切換。

 
  1.  
  2. import numpy as np

  3. a = np.array([1, 256, 8755], dtype = np.int16)

  4. print '我們的陣列是:'

  5. print a

  6. print '以十六進位制表示記憶體中的資料:'

  7. print map(hex,a)

  8. # byteswap() 函式通過傳入 true 來原地交換

  9. print '呼叫 byteswap() 函式:'

  10. print a.byteswap(True)

  11. print '十六進位制形式:'

  12. print map(hex,a)

  13. # 我們可以看到位元組已經交換了

  14.  

輸出如下:

 
  1. 我們的陣列是:

  2. [1 256 8755]

  3.  
  4. 以十六進位制表示記憶體中的資料:

  5. ['0x1', '0x100', '0x2233']

  6.  
  7. 呼叫 byteswap() 函式:

  8. [256 1 13090]

  9.  
  10. 十六進位制形式:

  11. ['0x100', '0x1', '0x3322']

  12.  

NumPy - 副本和檢視

在執行函式時,其中一些返回輸入陣列的副本,而另一些返回檢視。 當內容物理儲存在另一個位置時,稱為副本。 另一方面,如果提供了相同記憶體內容的不同檢視,我們將其稱為檢視

無複製

簡單的賦值不會建立陣列物件的副本。 相反,它使用原始陣列的相同id()來訪問它。 id()返回 Python 物件的通用識別符號,類似於 C 中的指標。

此外,一個陣列的任何變化都反映在另一個陣列上。 例如,一個陣列的形狀改變也會改變另一個陣列的形狀。

示例

 
  1.  
  2. import numpy as np

  3. a = np.arange(6)

  4. print '我們的陣列是:'

  5. print a

  6. print '呼叫 id() 函式:'

  7. print id(a)

  8. print 'a 賦值給 b:'

  9. b = a

  10. print b

  11. print 'b 擁有相同 id():'

  12. print id(b)

  13. print '修改 b 的形狀:'

  14. b.shape = 3,2

  15. print b

  16. print 'a 的形狀也修改了:'

  17. print a

  18.  

輸出如下:

 
  1.  
  2. 我們的陣列是:

  3. [0 1 2 3 4 5]

  4.  
  5. 呼叫 id() 函式:

  6. 139747815479536

  7.  
  8. a 賦值給 b:

  9. [0 1 2 3 4 5]

  10. b 擁有相同 id():

  11. 139747815479536

  12.  
  13. 修改 b 的形狀:

  14. [[0 1]

  15. [2 3]

  16. [4 5]]

  17.  
  18. a 的形狀也修改了:

  19. [[0 1]

  20. [2 3]

  21. [4 5]]

  22.  
  23.  

檢視或淺複製

NumPy 擁有ndarray.view()方法,它是一個新的陣列物件,並可檢視原始陣列的相同資料。 與前一種情況不同,新陣列的維數更改不會更改原始資料的維數。

示例

 
  1.  
  2. import numpy as np

  3. # 最開始 a 是個 3X2 的陣列

  4. a = np.arange(6).reshape(3,2)

  5. print '陣列 a:'

  6. print a

  7. print '建立 a 的檢視:'

  8. b = a.view()

  9. print b

  10. print '兩個陣列的 id() 不同:'

  11. print 'a 的 id():'

  12. print id(a)

  13. print 'b 的 id():'

  14. print id(b)

  15. # 修改 b 的形狀,並不會修改 a

  16. b.shape = 2,3

  17. print 'b 的形狀:'

  18. print b

  19. print 'a 的形狀:'

  20. print a

  21.  

輸出如下:

 
  1.  
  2. 陣列 a:

  3. [[0 1]

  4. [2 3]

  5. [4 5]]

  6.  
  7. 建立 a 的檢視:

  8. [[0 1]

  9. [2 3]

  10. [4 5]]

  11.  
  12. 兩個陣列的 id() 不同:

  13. a 的 id():

  14. 140424307227264

  15. b 的 id():

  16. 140424151696288

  17.  
  18. b 的形狀:

  19. [[0 1 2]

  20. [3 4 5]]

  21.  
  22. a 的形狀:

  23. [[0 1]

  24. [2 3]

  25. [4 5]]

  26.  
  27.  

陣列的切片也會建立檢視:

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[10,10], [2,3], [4,5]])

  4. print '我們的陣列:'

  5. print a

  6. print '建立切片:'

  7. s = a[:, :2]

  8. print s

  9.  

輸出如下:

 
  1.  
  2. 我們的陣列:

  3. [[10 10]

  4. [ 2 3]

  5. [ 4 5]]

  6.  
  7. 建立切片:

  8. [[10 10]

  9. [ 2 3]

  10. [ 4 5]]

  11.  
  12.  

深複製

ndarray.copy()函式建立一個深層副本。 它是陣列及其資料的完整副本,不與原始陣列共享。

示例

 
  1.  
  2. import numpy as np

  3. a = np.array([[10,10], [2,3], [4,5]])

  4. print '陣列 a:'

  5. print a

  6. print '建立 a 的深層副本:'

  7. b = a.copy()

  8. print '陣列 b:'

  9. print b

  10. # b 與 a 不共享任何內容

  11. print '我們能夠寫入 b 來寫入 a 嗎?'

  12. print b is a

  13. print '修改 b 的內容:'

  14. b[0,0] = 100

  15. print '修改後的陣列 b:'

  16. print b

  17. print 'a 保持不變:'

  18. print a

  19.  

輸出如下:

 
  1.  
  2. 陣列 a:

  3. [[10 10]

  4. [ 2 3]

  5. [ 4 5]]

  6.  
  7. 建立 a 的深層副本:

  8. 陣列 b:

  9. [[10 10]

  10. [ 2 3]

  11. [ 4 5]]

  12. 我們能夠寫入 b 來寫入 a 嗎?

  13. False

  14.  
  15. 修改 b 的內容:

  16. 修改後的陣列 b:

  17. [[100 10]

  18. [ 2 3]

  19. [ 4 5]]

  20.  
  21. a 保持不變:

  22. [[10 10]

  23. [ 2 3]

  24. [ 4 5]]

  25.  
  26.  

NumPy - 矩陣庫

NumPy 包包含一個 Matrix庫numpy.matlib。此模組的函式返回矩陣而不是返回ndarray物件。

matlib.empty()

matlib.empty()函式返回一個新的矩陣,而不初始化元素。 該函式接受以下引數。

 
  1.  
  2. numpy.matlib.empty(shape, dtype, order)

  3.  
  4.  

其中:

序號引數及描述
1.shape 定義新矩陣形狀的整數或整數元組
2.Dtype 可選,輸出的資料型別
3.order C 或者 F

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.empty((2,2))

  5. # 填充為隨機資料

  6.  

輸出如下:

 
  1. [[ 2.12199579e-314, 4.24399158e-314]

  2. [ 4.24399158e-314, 2.12199579e-314]]

  3.  

numpy.matlib.zeros()

此函式返回以零填充的矩陣。

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.zeros((2,2))

  5.  

輸出如下:

 
  1.  
  2. [[ 0. 0.]

  3. [ 0. 0.]])

  4.  
  5.  

numpy.matlib.ones()

此函式返回以一填充的矩陣。

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.ones((2,2))

  5.  

輸出如下:

 
  1. [[ 1. 1.]

  2. [ 1. 1.]]

  3.  

numpy.matlib.eye()

這個函式返回一個矩陣,對角線元素為 1,其他位置為零。 該函式接受以下引數。

 
  1.  
  2. numpy.matlib.eye(n, M,k, dtype)

  3.  
  4.  

其中:

序號引數及描述
1.n 返回矩陣的行數
2.M 返回矩陣的列數,預設為n
3.k 對角線的索引
4.dtype 輸出的資料型別

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.eye(n = 3, M = 4, k = 0, dtype = float)

  5.  

輸出如下:

 
  1.  
  2. [[ 1. 0. 0. 0.]

  3. [ 0. 1. 0. 0.]

  4. [ 0. 0. 1. 0.]])

  5.  
  6.  

numpy.matlib.identity()

numpy.matlib.identity()函式返回給定大小的單位矩陣。單位矩陣是主對角線元素都為 1 的方陣。

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.identity(5, dtype = float)

  5.  

輸出如下:

 
  1. [[ 1. 0. 0. 0. 0.]

  2. [ 0. 1. 0. 0. 0.]

  3. [ 0. 0. 1. 0. 0.]

  4. [ 0. 0. 0. 1. 0.]

  5. [ 0. 0. 0. 0. 1.]]

  6.  

numpy.matlib.rand()

·numpy.matlib.rand()`函式返回給定大小的填充隨機值的矩陣。

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4. print np.matlib.rand(3,3)

  5.  

輸出如下:

 
  1. [[ 0.82674464 0.57206837 0.15497519]

  2. [ 0.33857374 0.35742401 0.90895076]

  3. [ 0.03968467 0.13962089 0.39665201]]

  4.  

注意,矩陣總是二維的,而ndarray是一個 n 維陣列。 兩個物件都是可互換的。

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4.  
  5. i = np.matrix('1,2;3,4')

  6. print i

  7.  

輸出如下:

 
  1. [[1 2]

  2. [3 4]]

  3.  

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4.  
  5. j = np.asarray(i)

  6. print j

  7.  

輸出如下:

 
  1. [[1 2]

  2. [3 4]]

  3.  

示例

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4.  
  5. k = np.asmatrix (j)

  6. print k

  7.  

輸出如下:

 
  1. [[1 2]

  2. [3 4]]

  3.  

NumPy - 線性代數

NumPy 包包含numpy.linalg模組,提供線性代數所需的所有功能。 此模組中的一些重要功能如下表所述。

序號函式及描述
1.dot 兩個陣列的點積
2.vdot 兩個向量的點積
3.inner 兩個陣列的內積
4.matmul 兩個陣列的矩陣積
5.determinant 陣列的行列式
6.solve 求解線性矩陣方程
7.inv 尋找矩陣的乘法逆矩陣

numpy.dot()

此函式返回兩個陣列的點積。 對於二維向量,其等效於矩陣乘法。 對於一維陣列,它是向量的內積。 對於 N 維陣列,它是a的最後一個軸上的和與b的倒數第二個軸的乘積。

 
  1.  
  2. import numpy.matlib

  3. import numpy as np

  4.  
  5. a = np.array([[1,2],[3,4]])

  6. b = np.array([[11,12],[13,14]])

  7. np.dot(a,b)

  8.  

輸出如下:

 
  1. [[37 40]

  2. [85 92]]

要注意點積計算為:

[[1*11+2*13, 1*12+2*14],[3*11+4*13, 3*12+4*14]]

numpy.vdot()

此函式返回兩個向量的點積。 如果第一個引數是複數,那麼它的共軛複數會用於計算。 如果引數id是多維陣列,它會被展開。

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2],[3,4]])

  4. b = np.array([[11,12],[13,14]])

  5. print np.vdot(a,b)

  6.  

輸出如下:

130

注意:1*11 + 2*12 + 3*13 + 4*14 = 130

numpy.inner()

此函式返回一維陣列的向量內積。 對於更高的維度,它返回最後一個軸上的和的乘積。

例子

 
  1.  
  2. import numpy as np

  3. print np.inner(np.array([1,2,3]),np.array([0,1,0]))

  4. # 等價於 1*0+2*1+3*0

  5.  

輸出如下:

2

例子

 
  1.  
  2. # 多維陣列示例

  3. import numpy as np

  4. a = np.array([[1,2], [3,4]])

  5.  
  6. print '陣列 a:'

  7. print a

  8. b = np.array([[11, 12], [13, 14]])

  9.  
  10. print '陣列 b:'

  11. print b

  12.  
  13. print '內積:'

  14. print np.inner(a,b)

  15.  

輸出如下:

 
  1.  
  2. 陣列 a:

  3. [[1 2]

  4. [3 4]]

  5.  
  6. 陣列 b:

  7. [[11 12]

  8. [13 14]]

  9.  
  10. 內積:

  11. [[35 41]

  12. [81 95]]

  13.  

上面的例子中,內積計算如下:

 
  1. 1*11+2*12, 1*13+2*14

  2. 3*11+4*12, 3*13+4*14

numpy.matmul

numpy.matmul()函式返回兩個陣列的矩陣乘積。 雖然它返回二維陣列的正常乘積,但如果任一引數的維數大於2,則將其視為存在於最後兩個索引的矩陣的棧,並進行相應廣播。

另一方面,如果任一引數是一維陣列,則通過在其維度上附加 1 來將其提升為矩陣,並在乘法之後被去除。

例子

 
  1.  
  2. # 對於二維陣列,它就是矩陣乘法

  3. import numpy.matlib

  4. import numpy as np

  5.  
  6. a = [[1,0],[0,1]]

  7. b = [[4,1],[2,2]]

  8. print np.matmul(a,b)

  9.  

輸出如下:

 
  1. [[4 1]

  2. [2 2]]

例子

 
  1.  
  2. # 二維和一維運算

  3. import numpy.matlib

  4. import numpy as np

  5.  
  6. a = [[1,0],[0,1]]

  7. b = [1,2]

  8. print np.matmul(a,b)

  9. print np.matmul(b,a)

  10.  

輸出如下:

 
  1. [1 2]

  2. [1 2]

例子

 
  1.  
  2. # 維度大於二的陣列

  3. import numpy.matlib

  4. import numpy as np

  5.  
  6. a = np.arange(8).reshape(2,2,2)

  7. b = np.arange(4).reshape(2,2)

  8. print np.matmul(a,b)

  9.  

輸出如下:

 
  1. [[[2 3]

  2. [6 11]]

  3. [[10 19]

  4. [14 27]]]

numpy.linalg.det()

行列式線上性代數中是非常有用的值。 它從方陣的對角元素計算。 對於 2×2 矩陣,它是左上和右下元素的乘積與其他兩個的乘積的差。

換句話說,對於矩陣[[a,b],[c,d]],行列式計算為ad-bc。 較大的方陣被認為是 2×2 矩陣的組合。

numpy.linalg.det()函式計算輸入矩陣的行列式。

例子

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,2], [3,4]])

  4. print np.linalg.det(a)

  5.  

輸出如下:

 
  1.  
  2. -2.0

  3.  

例子

 
  1.  
  2. b = np.array([[6,1,1], [4, -2, 5], [2,8,7]])

  3. print b

  4. print np.linalg.det(b)

  5. print 6*(-2*7 - 5*8) - 1*(4*7 - 5*2) + 1*(4*8 - -2*2)

  6.  

輸出如下:

 
  1. [[ 6 1 1]

  2. [ 4 -2 5]

  3. [ 2 8 7]]

  4.  
  5. -306.0

  6.  
  7. -306

numpy.linalg.solve()

numpy.linalg.solve()函式給出了矩陣形式的線性方程的解。

考慮以下線性方程:

 
  1. x + y + z = 6

  2.  
  3. 2y + 5z = -4

  4.  
  5. 2x + 5y - z = 27

可以使用矩陣表示為:

如果矩陣成為AXB,方程變為:

AX = B  

X = A^(-1)B 

numpy.linalg.inv()

我們使用numpy.linalg.inv()函式來計算矩陣的逆。 矩陣的逆是這樣的,如果它乘以原始矩陣,則得到單位矩陣。

例子

 
  1.  
  2. import numpy as np

  3.  
  4. x = np.array([[1,2],[3,4]])

  5. y = np.linalg.inv(x)

  6. print x

  7. print y

  8. print np.dot(x,y)

  9.  

輸出如下:

 
  1. [[1 2]

  2. [3 4]]

  3. [[-2. 1. ]

  4. [ 1.5 -0.5]]

  5. [[ 1.00000000e+00 1.11022302e-16]

  6. [ 0.00000000e+00 1.00000000e+00]]

例子

現在讓我們在示例中建立一個矩陣A的逆。

 
  1.  
  2. import numpy as np

  3. a = np.array([[1,1,1],[0,2,5],[2,5,-1]])

  4.  
  5. print '陣列 a:'

  6. print a

  7. ainv = np.linalg.inv(a)

  8.  
  9. print 'a 的逆:'

  10. print ainv

  11.  
  12. print '矩陣 b:'

  13. b = np.array([[6],[-4],[27]])

  14. print b

  15.  
  16. print '計算:A^(-1)B:'

  17. x = np.linalg.solve(a,b)

  18. print x

  19. # 這就是線性方向 x = 5, y = 3, z = -2 的解

  20.  

輸出如下:

 
  1.  
  2. 陣列 a:

  3. [[ 1 1 1]

  4. [ 0 2 5]

  5. [ 2 5 -1]]

  6.  
  7. a 的逆:

  8. [[ 1.28571429 -0.28571429 -0.14285714]

  9. [-0.47619048 0.14285714 0.23809524]

  10. [ 0.19047619 0.14285714 -0.0952381 ]]

  11.  
  12. 矩陣 b:

  13. [[ 6]

  14. [-4]

  15. [27]]

  16.  
  17. 計算:A^(-1)B:

  18. [[ 5.]

  19. [ 3.]

  20. [-2.]]

  21.  

結果也可以使用下列函式獲取

x = np.dot(ainv,b)

NumPy - Matplotlib

Matplotlib 是 Python 的繪相簿。 它可與 NumPy 一起使用,提供了一種有效的 MatLab 開源替代方案。 它也可以和圖形工具包一起使用,如 PyQt 和 wxPython。

Matplotlib 模組最初是由 John D. Hunter 編寫的。 自 2012 年以來,Michael Droettboom 是主要開發者。 目前,Matplotlib 1.5.1 是可用的穩定版本。 該軟體包可以二進位制分發,其原始碼形式在 www.matplotlib.org 上提供。

通常,通過新增以下語句將包匯入到 Python 指令碼中:

 
  1.  
  2. from matplotlib import pyplot as plt

  3.  
  4.  

這裡pyplot()是 matplotlib 庫中最重要的函式,用於繪製 2D 資料。 以下指令碼繪製方程y = 2x + 5

示例

 
  1.  
  2. import numpy as np

  3. from matplotlib import pyplot as plt

  4.  
  5. x = np.arange(1,11)

  6. y = 2 * x + 5

  7. plt.title("Matplotlib demo")

  8. plt.xlabel("x axis caption")

  9. plt.ylabel("y axis caption")

  10. plt.plot(x,y) plt.show()

  11.  

ndarray物件xnp.arange()函式建立為x軸上的值。y軸上的對應值儲存在另一個陣列物件y中。 這些值使用matplotlib軟體包的pyplot子模組的plot()函式繪製。

圖形由show()函式展示。

上面的程式碼應該產生以下輸出:

 

Matplotlib Demo

作為線性圖的替代,可以通過向plot()函式新增格式字串來顯示離散值。 可以使用以下格式化字元。

字元描述
'-'實線樣式
'--'短橫線樣式
'-.'點劃線樣式
':'虛線樣式
'.'點標記
','畫素標記
'o'圓標記
'v'倒三角標記
'^'正三角標記
'<'左三角標記
'>'右三角標記
'1'下箭頭標記
'2'上箭頭標記
'3'左箭頭標記
'4'右箭頭標記
's'正方形標記
'p'五邊形標記
'*'星形標記
'h'六邊形標記 1
'H'六邊形標記 2
'+'加號標記
'x'X 標記
'D'菱形標記
'd'窄菱形標記
`''`豎直線標記
'_'水平線標記

還定義了以下顏色縮寫。

字元顏色
'b'藍色
'g'綠色
'r'紅色
'c'青色
'm'品紅色
'y'黃色
'k'黑色
'w'白色

要顯示圓來代表點,而不是上面示例中的線,請使用ob作為plot()函式中的格式字串。

示例

 
  1.  
  2. import numpy as np

  3. from matplotlib import pyplot as plt

  4.  
  5. x = np.arange(1,11)

  6. y = 2 * x + 5

  7. plt.title("Matplotlib demo")

  8. plt.xlabel("x axis caption")

  9. plt.ylabel("y axis caption")

  10. plt.plot(x,y,"ob")

  11. plt.show()

  12.  

上面的程式碼應該產生以下輸出:

 

Color Abbreviation

繪製正弦波

以下指令碼使用 matplotlib 生成正弦波圖

示例

 
  1.  
  2. import numpy as np

  3. import matplotlib.pyplot as plt

  4. # 計算正弦曲線上點的 x 和 y 座標

  5. x = np.arange(0, 3 * np.pi, 0.1)

  6. y = np.sin(x)

  7. plt.title("sine wave form")

  8. # 使用 matplotlib 來繪製點

  9. plt.plot(x, y)

  10. plt.show()

  11.  

 

Sine Wave

subplot()

subplot()函式允許你在同一圖中繪製不同的東西。 在下面的指令碼中,繪製正弦餘弦值。

示例

 
  1.  
  2. import numpy as np

  3. import matplotlib.pyplot as plt

  4. # 計算正弦和餘弦曲線上的點的 x 和 y 座標

  5. x = np.arange(0, 3 * np.pi, 0.1)

  6. y_sin = np.sin(x)

  7. y_cos = np.cos(x)

  8. # 建立 subplot 網格,高為 2,寬為 1

  9. # 啟用第一個 subplot

  10. plt.subplot(2, 1, 1)

  11. # 繪製第一個影像

  12. plt.plot(x, y_sin)

  13. plt.title('Sine')

  14. # 將第二個 subplot 啟用,並繪製第二個影像

  15. plt.subplot(2, 1, 2)

  16. plt.plot(x, y_cos)

  17. plt.title('Cosine')

  18. # 展示影像

  19. plt.show()

  20.  

上面的程式碼應該產生以下輸出:

 

Sub Plot

bar()

pyplot子模組提供bar()函式來生成條形圖。 以下示例生成兩組xy陣列的條形圖。

示例

 
  1.  
  2. from matplotlib import pyplot as plt

  3. x = [5,8,10]

  4. y = [12,16,6]

  5. x2 = [6,9,11]

  6. y2 = [6,15,7]

  7. plt.bar(x, y, align = 'center')

  8. plt.bar(x2, y2, color = 'g', align = 'center')

  9. plt.title('Bar graph')

  10. plt.ylabel('Y axis')

  11. plt.xlabel('X axis')

  12. plt.show()

  13.  

NumPy - 使用 Matplotlib 繪製直方圖

NumPy 有一個numpy.histogram()函式,它是資料的頻率分佈的圖形表示。 水平尺寸相等的矩形對應於類間隔,稱為bin,變數height對應於頻率。

numpy.histogram()

numpy.histogram()函式將輸入陣列和bin作為兩個引數。 bin陣列中的連續元素用作每個bin的邊界。

 
  1.  
  2. import numpy as np

  3.  
  4. a = np.array([22,87,5,43,56,73,55,54,11,20,51,5,79,31,27]) ]

  5. np.histogram(a,bins = [0,20,40,60,80,100])

  6. hist,bins = np.histogram(a,bins = [0,20,40,60,80,100])

  7. print hist

  8. print bins

  9.  

輸出如下:

 
  1. [3 4 5 2 1]

  2. [0 20 40 60 80 100]

  3.  

plt()

Matplotlib 可以將直方圖的數字表示轉換為圖形。 pyplot子模組的plt()函式將包含資料和bin陣列的陣列作為引數,並轉換為直方圖。

 
  1.  
  2. from matplotlib import pyplot as plt

  3. import numpy as np

  4.  
  5. a = np.array([22,87,5,43,56,73,55,54,11,20,51,5,79,31,27])

  6. plt.hist(a, bins = [0,20,40,60,80,100])

  7. plt.title("histogram")

  8. plt.show()

  9.  

輸出如下:

 

Histogram Plot

NumPy - IO

ndarray物件可以儲存到磁碟檔案並從磁碟檔案載入。 可用的 IO 功能有:

  • load()save()函式處理 numPy 二進位制檔案(帶npy副檔名)

  • loadtxt()savetxt()函式處理正常的文字檔案

NumPy 為ndarray物件引入了一個簡單的檔案格式。 這個npy檔案在磁碟檔案中,儲存重建ndarray所需的資料、圖形、dtype和其他資訊,以便正確獲取陣列,即使該檔案在具有不同架構的另一臺機器上。

numpy.save()

numpy.save()檔案將輸入陣列儲存在具有npy副檔名的磁碟檔案中。

 
  1.  
  2. import numpy as np

  3. a = np.array([1,2,3,4,5])

  4. np.save('outfile',a)

  5.  

為了從outfile.npy重建陣列,請使用load()函式。

 
  1.  
  2. import numpy as np

  3. b = np.load('outfile.npy')

  4. print b

  5.  

輸出如下:

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

  3.  
  4.  

save()load()函式接受一個附加的布林引數allow_pickles。 Python 中的pickle用於在儲存到磁碟檔案或從磁碟檔案讀取之前,對物件進行序列化和反序列化。

savetxt()

以簡單文字檔案格式儲存和獲取陣列資料,是通過savetxt()loadtx()函式完成的。

示例

 
  1.  
  2. import numpy as np

  3.  
  4. a = np.array([1,2,3,4,5])

  5. np.savetxt('out.txt',a)

  6. b = np.loadtxt('out.txt')

  7. print b

  8.  

輸出如下:

 
  1. [ 1. 2. 3. 4. 5.]

  2.  

savetxt()loadtxt()數接受附加的可選引數,例如頁首,頁尾和分隔符。

NumPy - 實用資源

以下資源包含有關 NumPy 的其他資訊。 請使用它們獲得更多的深入知識。


轉載連結:https://www.jianshu.com/p/57e3c0a92f3a

相關文章