zubinmehta
zubinmehta

Reputation: 4533

How can i store a list within a numpy 2d array?

How can i store a list within a numpy 2d array?

import numpy  
A = numpy.empty([5,5], dtype=**?**)  

What should be the dtype for a variable list kind of thing here? or Is there a different way to implement this, which I strongly feel would be the case.

Upvotes: 4

Views: 8118

Answers (1)

Bogdan
Bogdan

Reputation: 8246

I think what you want is:

>>> input = numpy.array(range(10))
>>> data = numpy.zeros((2,2), dtype=numpy.ndarray)
>>> data[1][1] = input
>>> data
    array([[0, 0],
          [0, [0 1 2 3 4 5 6 7 8 9]]], dtype=object)

Upvotes: 6

Related Questions