numpy中flatten和ravel函数有什么区别?

时间:2015-03-08 18:49:44

标签: python numpy multidimensional-array flatten numpy-ndarray

import numpy as np
y = np.array(((1,2,3),(4,5,6),(7,8,9)))
OUTPUT:
print(y.flatten())
[1   2   3   4   5   6   7   8   9]
print(y.ravel())
[1   2   3   4   5   6   7   8   9]

两个函数都返回相同的列表。 那么两个不同的功能需要执行相同的工作。

3 个答案:

答案 0 :(得分:303)

目前的API是:

  • flatten始终返回副本。
  • ravel尽可能返回原始数组的视图。这在打印输出中不可见,但是如果修改ravel返回的数组,它可能会修改原始数组中的条目。如果修改从flatten返回的数组中的条目,则永远不会发生这种情况。 ravel通常会更快,因为没有复制内存,但你必须更加小心修改它返回的数组。
  • 只要数组的步幅允许,
  • reshape((-1,))就会获得一个视图,即使这意味着你不会总是得到一个连续的数组。

答案 1 :(得分:31)

正如here解释的那样,一个关键的区别是flatten是一个ndarray对象的方法,因此只能为真正的numpy数组调用。相反,ravel()是一个库级函数,因此可以在任何可以成功解析的对象上调用。例如,ravel()将在ndarrays列表上工作,而flatten不适用于该类型的对象。

@IanH还在答案中指出了与记忆处理的重要区别。

答案 2 :(得分:6)

这是函数的正确名称空间:

这两个函数都返回指向新内存结构的展平一维数组。

import numpy
a = numpy.array([[1,2],[3,4]])

r = numpy.ravel(a)
f = numpy.ndarray.flatten(a)  

print(id(a))
print(id(r))
print(id(f))

print(r)
print(f)

print("\nbase r:", r.base)
print("\nbase f:", f.base)

---returns---
140541099429760
140541099471056
140541099473216

[1 2 3 4]
[1 2 3 4]

base r: [[1 2]
 [3 4]]

base f: None

在上面的示例中:

  • 结果的存储位置不同,
  • 结果看起来一样
  • flatten将返回副本
  • ravel将返回一个视图。

我们如何检查某物是否是副本? 使用.base的{​​{1}}属性。如果是视图,则基础将是原始数组;如果是副本,则基数为ndarray