numpy中的ravel()方法介绍:不论矩阵的维度是多少。使用ravel()可以让矩阵统统变为一维数组。
话不多说,直接上代码:
import numpy as np # 导入numpy模块
a, b = np.mgrid[1:4:1, 2:3:1] # 用mgrid()方法生成等差数组a,b
#
print("a:\n", a, "\n", "b:\n", b) # 打印a,b
Aftera = a.ravel() # 用ravel()方法将数组a拉成一维数组
Afterb = b.ravel() # 用ravel()方法将数组b拉成一维数组
print("Aftera:", Aftera) # 打印a
print("Afterb:", Afterb) # 打印b
sum = Aftera + Afterb
print(sum)
运行结果: