python - 在python中更改numpy数组的格式

Question

我有一个 (2x2)numpy数组，其中包含一个 (1x3) 列表，如下所示：

[[ 1, 2, 3], [ 4, 5, 6]
 [ 7, 8, 9], [10,11,12]]

我想把它分解成像这个（2x6）数组（或矩阵，我无法区分）这样的单个元素：

[ 1, 2, 3,  4,  5,  6]
[ 7, 8, 9, 10, 11, 12]

我可以通过处理第一个数组的每个元素并将其放到第二个数组中来创建第二个数组。numpy但是我想知道图书馆是否提供了一种更简单的方法。

标签： pythonnumpynumpy-ndarray

Answer 1

你可以简单地使用调整大小功能来做到这一点

import numpy as np
a=[[ 1, 2, 3], [ 4, 5, 6],
 [ 7, 8, 9], [10,11,12]]
a = numpy.array(a)
a.resize(2,6)
print(a)

输出

array([[ 1,  2,  3,  4,  5,  6],
       [ 7,  8,  9, 10, 11, 12]])