[Python] 扁平化函数 ravel()和flatten()

功能:

二者都是将n×m的矩阵,变为1×mn的行向量。

建议使用flatten(),而不建议使用ravel()
因为在使用过程中flatten()分配了新的内存,但ravel()返回的是一个数组的视图。
ravel()返回的视图,类似对数组的引用(说引用不太恰当,因为原数组和ravel()返回的数组的地址并不一样),故在使用过程会修改原数组。

示例:

import numpy as np

a = arange(12).reshape(3,4)
print(a)
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]

# 创建一个和a相同内容的数组b
b = a.copy()

# 扁平化处理
c = a.ravel()
d = b.flatten()

# 输出c和d数组
# c和d数组都是扁平化后的数组,具有相同的内容
print(c)
# [ 0  1  2  3  4  5  6  7  8  9 10 11]
print(d)
# [ 0  1  2  3  4  5  6  7  8  9 10 11]

# a,b,c,d是四个不同的对象
print(a is c)
# False
print(b is d)
# False


# c 由 a 经过 ravel() 得到。再修改c的时候,a也被修改。
c[1] = 99
d[1] = 99
print(a)
# [[ 0 99  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]
print(b)
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]
print(c)
# [ 0 99  2  3  4  5  6  7  8  9 10 11]
print(d)
# [ 0 99  2  3  4  5  6  7  8  9 10 11]

你可能感兴趣的:(Python,Python)