NumPy - 矩陣的計算
1. 介紹
在Python中,常用的列表(list)或者array模塊,對於計算矩陣太花費時間空間了,需要有個專門處理數字矩陣的工具,於是NumPy提供了一些方法。
2. 創建矩陣
numpy的創建矩陣原型如下
numpy.array(object, dtype = None, copy = True, order = None, subok = False, ndmin = 0)
import numpy as np
a = np.array([1, 2, 3, 4]) #創建1維陣列
b = np.array([[1, 2, 3, 4],[2, 6, 6, 2], [7, 4, 1, 2]]) #創建3x4陣列
print(a[0], a[1], a[2], a[3]) #印出 "1 2 3 4"
print(a) #印出 "[1, 2, 3, 4]"
print(a.shape) #a陣列的大小為"(4,)"
print(b.shape) #a陣列的大小為"(3, 4)"
c = np.array([1, 2, 3], dtype = complex) #創建複數類型陣列
用shape方法可以看出矩陣的維度,若要改變它的維度,或是說改變它的軸,重新排列可以使用reshape方法
c = a.reshape((2,2))
print(c) #印出 "[[1, 2],
[3, 4]]"