Numpy array as single element of pandas dataframe

Question

I have a numpy array and an empty dataframe:

element = numpy.array([1,2,3])
df = pandas.DataFrame(columns = ["Col"])

I want to insert element in the first row of df. The following code:

df["Col"] = element

Gives me a dataframe 3x1 whose elements are 1, 2 and 3. I want a dataframe 1x1 whose element is the array. How can I get this result? Thanks in advance!

jezrael · Accepted Answer · 2021-05-20 11:56:34Z

3

Use DataFrame.loc or DataFrame.at for specify label for set array to DataFrame:

df.loc[0, "Col"] = element
print (df)
         Col
0  [1, 2, 3]

df.at[0, "Col"] = element

answered May 20, 2021 at 11:56

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Comments

timgeb · Accepted Answer · 2021-05-20 12:21:28Z

2

Wrap element in a list.

>>> df['Col'] = [element]
>>> df
         Col
0  [1, 2, 3]

answered May 20, 2021 at 12:21

timgeb

79.2k20 gold badges129 silver badges150 bronze badges

Collectives™ on Stack Overflow

Numpy array as single element of pandas dataframe

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related