1. 程式人生 > >python numpy基本用法

python numpy基本用法

1.陣列用法

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

這樣就可以對array_1進行計算了。

獲得二維陣列

array_3 = np.array([list_1,list_2])

獲得陣列的型別

array_3.dtype

獲得陣列的元素個數

array_3.size

獲得陣列行列數

array_3.shape

構造陣列並只要單數

rray_4 = np.arange(1,10,2)

獲得正交陣列

np.eye(5)

陣列索引方式

b[1][1]

獲得一個隨機矩陣

a = np.mat(np.random.randint(10,size = 20).reshape(4,5))
matrix([[8, 7, 4, 6, 1],
        [5, 5, 1, 1, 8],
        [4, 6, 3, 7, 8],
        [4, 4, 7, 8, 9]])

矩陣的用法

a + b
a - b
a * b

將陣列轉為矩陣

A = np.mat(a)

將陣列存入檔案並讀取

import pickle
import numpy as np
x = np.arange(10)
f = open('x.pkl','wb')
pickle.dump(x,f)
ls#用以檢視是否建立檔案成功
f = open('x.pkl','rb')
pickle.load(f)#讀取了檔案內容
np.save('one_array',x)#測試將陣列存入另外一個檔案
np.load('one_array.npy')

存取兩個陣列

x = np.arange(10)
y = np.arange(10)
np.savez('two_array.npz',a=x,b=y)
c = np.load('two_array.npz')
c['a']
c['b']