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']