CODEWITHSUNDEEP
pythonpandasdataframe
Mainland
Mainland

Reputation: 4554

Python appending a list to dataframe as element

My dataframe is given below

df = 

index    data1
0         20
1         30
2         40

I want to add a new column and each element consiting a list.

My expected output is

df = 

index    data1    list_data
0         20       [200,300,90]
1         30       [200,300,90,78,90]
2         40       [1200,2300,390,89,78]

My present code: 

        df['list_data'] = []
        df['list_data'].loc[0] = [200,300,90]

Present output: 

    raise ValueError('Length of values does not match length of index')

ValueError: Length of values does not match length of index

Upvotes: 1

Views: 66

Answers (2)

Swati Srivastava
Swati Srivastava

Reputation: 1157

You can use pd.Series for your problem

import pandas as pd
lis = [[200, 300, 90], [200, 300, 90, 78, 90], [1200, 2300, 390, 89, 78]]
lis = pd.Series(lis)
df['list_data'] = lis

This gives the following output

   index   data1    list_data
0   0        20    [200, 300, 90]
1   1        30    [200, 300, 90, 78, 90]
2   2        40    [1200, 2300, 390, 89, 78]

Upvotes: 2

U13-Forward
U13-Forward

Reputation: 71560

Try using loc this way:

df['list_data'] = ''
df.loc[0, 'list_data'] = [200,300,90]

Upvotes: 1

Related Questions