将列表转换为数组时减少维度

Question

我想在将数组转换为列表后减小其维度

a = np.array([[1,2],[3,4]])
print a.shape
b = np.array([[1],[3,4]])
print b.shape

输出：

(2, 2)
(2,)

我希望 a 具有与 b 相同的形状，即 (2,)

Answer 1

>>> a = np.array([[1,2],[3,4], None])[:2]
>>> a
array([[1, 2], [3, 4]], dtype=object)
>>> a.shape
(2,)

有效，但可能是错误的做法（我是个麻木的新手）。

Answer 2

你知道b是什么吗？

b = np.array([[1],[3,4]])
print(repr(b))

array([[1], [3, 4]], dtype=object)

b 是一个包含 2 个元素的一维数组，每个元素都是一个列表。 np.array 这样做是因为 2 个子列表的长度不同，所以无法创建二维数组。

a = np.array([[1,2],[3,4]])
print(repr(a)) 

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

这里的 2 个子列表长度相同，因此可以创建一个二维数组。每个元素都是一个整数。 np.array 尝试创建输入允许的最高维数组。

创建像b 这样的另一个数组的最佳方法可能是制作一个副本，然后插入所需的列表。

a1 = b.copy()
a1[0] = [1,2]
# a1[1] = [3,4]
print(repr(a1))

array([[1, 2], [3, 4]], dtype=object)

你必须使用这种复杂的方法，因为你试图做一些“不自然”的事情。

您对使用vstack 发表评论。两者都有效：

In [570]: np.vstack((a,b))   # (3,2) array
Out[570]: 
array([[1, 2],
       [3, 4],
       [[1], [3, 4]]], dtype=object)

In [571]: np.vstack((a1,b))   # (2,2) array
Out[571]: 
array([[[1, 2], [3, 4]],
       [[1], [3, 4]]], dtype=object)

您的数组b 只不过是数组包装器中的原始列表。这真的是你需要的吗？ 2d a 是一个普通的 numpy 数组。 b 是一个奇怪的结构。