如何制作一个可变行大小的多维numpy数组？

Question

我想创建一个二维numpy数组，其中每行有不同数量的元素。

尝试

cells = numpy.array([[0,1,2,3], [2,3,4]])

出现错误

ValueError: setting an array element with a sequence.

Answer 1

这在Numpy中不太受支持（根据定义，几乎在任何地方，“二维数组”的所有行都是等长的）。Numpy数组的Python列表可能是一个很好的解决方案，因为这样您就可以获得Numpy的优点，在这里您可以使用它们：

cells = [numpy.array(a) for a in [[0,1,2,3], [2,3,4]]]