本文主要列出numpy模块常用方法
创建矩阵(采用ndarray对象)
对于python中的numpy模块,一般用其提供的ndarray对象。
创建一个ndarray对象很简单,只要将一个list作为参数即可。
import numpy as np
# 创建一维的narray对象
a = np.array([1,2,3,4,5])
# 创建二维的narray对象
a2 = np.array([[1,2,3,4,5],[6,7,8,9,10]])
# 创建多维对象以其类推
获取矩阵行数列数(二维情况)
要对矩阵进行遍历时,一般先获取矩阵的行数和列数。要获取narray对象的各维的长度,可以通过narray对象的shape属性
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a.shape) # 结果返回一个tuple元组 (2, 5)
print(a.shape[0]) # 获得行数,返回 2
print(a.shape[1]) # 获得列数,返回 5
矩阵的截取
按行列截取
矩阵的截取和list相同,可以通过[](方括号)来截取
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a[0:1]) # 截取第一行,返回 [[1,2,3,4,5]]
print(a[1,2:5]) # 截取第二行,第三、四列,返回 [8,9]
print(a[1,:]) # 截取第二行,返回 [ 6,7,8,9,10]
print(a[1:,2:]) # 截取第一行之后,第2列之后内容,返回[8,9,10]
按条件截取
按条件截取其实是在[](方括号)中传入自身的布尔语句
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
b = a[a>6] # 截取矩阵a中大于6的元素,范围的是一维数组
print(b) # 返回 [ 7 8 9 10]
# 其实布尔语句首先生成一个布尔矩阵,将布尔矩阵传入[](方括号)实现截取
print(a>6)
# 返回
[[False False False False False]
[False True True True True]]
按条件截取应用较多的是对矩阵中满足一定条件的元素变成特定的值。
例如:将矩阵中大于6的元素变成0。
import numpy as np
a = np.array([[1,2,3,4,5],[6,7,8,9,10]])
print(a)
#开始矩阵为
[[ 1 2 3 4 5]
[ 6 7 8 9 10]]
a[a>6] = 0
print(a)
#大于6清零后矩阵为
[[1 2 3 4 5]
[6 0 0 0 0]]
矩阵的合并
矩阵的合并可以通过numpy中的hstack方法和vstack方法实现
import numpy as np
a1 = np.array([[1,2],[3,4]])
a2 = np.array([[5,6],[7,8]])
# 注意! 参数传入时要以列表list或元组tuple的形式传入
print(np.hstack([a1,a2]))
# 横向合并,返回结果如下
[[1 2 5 6]
[3 4 7 8]]
print(np.vstack((a1,a2)))
# 纵向合并,返回结果如下
[[1 2]
[3 4]
[5 6]
[7 8]]
# 矩阵的合并也可以通过concatenatef方法。
np.c