2.4.2　NumPy模塊

Python中有一些數據類型，例如列表、字典、集合等，可以用于保存一系列的數據信息。不過對于大規模的數據計算，它們并不都適用。在AI領域的計算中，大規模數據一般以向量形式保存。NumPy模塊是專門用于向量計算的模塊。讓我們看看下面的例子。

首先建立一個列表，再建立一個數組，輸出并觀察二者的類型，可以看到二者的類型是不一樣的。x_list是列表（list）類型的，x_array是NumPy數組類型的。

import numpy as np
x_list=[1,2,3,4]
type(x_list)

list 

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

numpy.ndarray

NumPy數組類型的重要特點在于支持向量化計算，例如我們要計算數值的平方再求和，如果使用列表類型，需要寫一個循環，或者使用列表解析來完成。

x_sum=0
for x in x_list:
    x_sum=x_sum+x**2
print(x_sum)

30

sum([x**2 for x in x_list])

30 

如果使用NumPy數組類型，則更加簡單，它可以直接對每個數值做平方運算。通過這個例子可以看到NumPy數組類型的優點，它的向量化計算非常方便、快速。

sum(x_array**2)

30