数组中的Numpy数组,长度不等

时间:2017-11-02 17:51:05

标签: arrays numpy

我想知道如何在numpy数组中创建一个numpy数组。主阵列中的每个阵列都应该具有不同的长度。

主数组应该是

中数组对象的索引号

我已经有了这种类型数组的示例,但我想重新创建类似的

示例代码:

test.shape

(200)

测试[0] .shape

(5,10)

测试[1] .shape

(3,10)

测试[2] .shape

(6,10)

2 个答案:

答案 0 :(得分:0)

您需要使用dtype=object创建一个数组,以便numpy知道将每个条目视为Python对象,而不是将整个嵌套列表视为单个数组。

例如:

import numpy as np
x = np.empty((5, 10))
y = np.empty((3, 10))
z = np.empty((6, 10))

test = np.array([x, y, z], dtype=object)
test[0].shape
# [5, 10]

答案 1 :(得分:0)

创建对象数组最可靠的方法是初始化并填充它。 np.array的行为太多了。

In [658]: alist = [np.ones((5,10),int), np.zeros((3,10),int), np.arange(60).resh
     ...: ape(6,10)]
In [659]: arr = np.empty(len(alist), dtype=object)
In [660]: arr[:] = alist
In [661]: arr
Out[661]: 
array([ array([[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]),
       array([[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]),
       array([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14, 15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24, 25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34, 35, 36, 37, 38, 39],
       [40, 41, 42, 43, 44, 45, 46, 47, 48, 49],
       [50, 51, 52, 53, 54, 55, 56, 57, 58, 59]])], dtype=object)

np.array的行为随着组件的相对形状而变化:

漂亮的对象数组:

In [668]: np.array((np.ones((3,5),int), np.ones((2,5),int)),object)
Out[668]: 
array([array([[1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1]]),
       array([[1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1]])], dtype=object)

3d数组:

In [669]: np.array((np.ones((3,5),int), np.ones((3,5),int)),object)
Out[669]: 
array([[[1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1]],

       [[1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1]]], dtype=object)

错误

In [670]: np.array((np.ones((3,4),int), np.ones((3,5),int)),object)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-670-e9b41001d868> in <module>()
----> 1 np.array((np.ones((3,4),int), np.ones((3,5),int)),object)

ValueError: could not broadcast input array from shape (3,4) into shape (3)