我想知道如何在numpy数组中创建一个numpy数组。主阵列中的每个阵列都应该具有不同的长度。
主数组应该是
中数组对象的索引号我已经有了这种类型数组的示例,但我想重新创建类似的
示例代码:
test.shape
(200)
测试[0] .shape
(5,10)
测试[1] .shape
(3,10)
测试[2] .shape
(6,10)
答案 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)