Adding arrays to dataframe column

Question

Let's assume I have this dataframe df:

   'Location'  'Rec ID'  'Duration'
0     Houston       126          17
1     Chicago       338        19.3

I would like to add a column with arrays corresponding to my recordings like:

   'Location'  'Rec ID'  'Duration'                           'Rec'
0     Houston       126          17    [0.2, 0.34, 0.45, ..., 0.28]
1     Chicago       338        19.3    [0.12, 0.3, 0.41, ..., 0.39]

When I do the df.set_value() command I get the following error:

ValueError: setting an array element with a sequence.

Take alook stackoverflow.com/questions/18646076/…

Hisham Karam
– Hisham Karam

2016-10-14 05:14:32 +00:00
Commented Oct 14, 2016 at 5:14 — Hisham Karam
– Hisham Karam, Commented Oct 14, 2016 at 5:14

jezrael · Accepted Answer · 2016-10-14 05:41:21Z

1

I think the easiest is to assign list of lists, only you need same length of lists as length of DataFrame:

arr = [[0.2, 0.34, 0.45, 0.28], [0.12, 0.3, 0.41, 0.39]]
print (arr)
[[0.2, 0.34, 0.45, 0.28], [0.12, 0.3, 0.41, 0.39]]

print (len(arr))
2
print (len(df))
2

df["'Rec'"] = arr
print (df)
   'Location'     'Rec  ID'  'Duration'                    'Rec'
0           0  Houston  126        17.0  [0.2, 0.34, 0.45, 0.28]
1           1  Chicago  338        19.3  [0.12, 0.3, 0.41, 0.39]

If use numpy array, first convert tolist:

arr = np.array([[0.2, 0.34, 0.45, 0.28], [0.12, 0.3, 0.41, 0.39]])
print (arr)
[[ 0.2   0.34  0.45  0.28]
 [ 0.12  0.3   0.41  0.39]]

df["'Rec'"] = arr.tolist()

print (df)
   'Location'     'Rec  ID'  'Duration'                    'Rec'
0           0  Houston  126        17.0  [0.2, 0.34, 0.45, 0.28]
1           1  Chicago  338        19.3  [0.12, 0.3, 0.41, 0.39]

edited Oct 14, 2016 at 5:41

answered Oct 14, 2016 at 5:19

jezrael

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

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

1 Comment

jezrael Over a year ago

@piRSquared - Thank you. I dont test it.

Collectives™ on Stack Overflow

Adding arrays to dataframe column

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related