python - 如何拆分/ reshape 一个 numpy 数组

Question

我有一个名为“results”的 numpy 数组，它是这样定义的

array([1, 2, 3, 4, 5, 6])

但我需要它看起来像这样:

array([1, 2], [3, 4], [5, 6])

如何在 Python 中将“结果”转换为这个新数组？我最终得到的数组仍然需要是一个 numpy 数组。

Answer 1

您可以使用 reshape 方法直接实现这一点。

例如:

In [1]: import numpy as np

In [2]: arr = np.array([1, 2, 3, 4, 5, 6])

In [3]: reshaped = arr.reshape((3, 2))

In [4]: reshaped
Out[4]: 
array([[1, 2],
       [3, 4],
       [5, 6]])

---

请注意，在可能的情况下，reshape 将为您提供数组的 View 。换句话说，您正在查看与原始数组相同的基础数据:

In [5]: reshaped[0][0] = 7

In [6]: reshaped
Out[6]: 
array([[7, 2],
       [3, 4],
       [5, 6]])

In [7]: arr
Out[7]: array([7, 2, 3, 4, 5, 6])

这几乎总是一个优势。但是，如果您不希望出现这种情况，您可以随时复制一份:

In [8]: copy = np.copy(reshaped)

In [9]: copy[0][0] = 9

In [10]: copy
Out[10]: 
array([[9, 2],
       [3, 4],
       [5, 6]])

In [11]: reshaped
Out[11]: 
array([[7, 2],
       [3, 4],
       [5, 6]])