Add multiple data in columns using indexes pandas

Question

Bellow are the steps on how I would like to get my data saved:

Step 1

ID	A	B
1	rand
4	rand

Step 2

ID	A	B
1	rand	rand
4	rand	rand

Step 3

ID	A	B
1	rand	rand
4	rand	rand
3	rand
6	rand

Step 4

ID	A	B
1	rand	rand
4	rand	rand
3	rand	rand
6	rand	rand

Current code:

from numpy.random import rand
import pandas as pd
IDS = [[1,4],[3,6]]
raws = ['A','B']
df = pd.DataFrame(columns=['ID']+raws)
print(df)

for ind in IDS:
    df.index = ind
    for raw in raws:
        df[ind][raw] = rand(len(ind))

Issue: At the moment I have the following error: "ValueError: Length mismatch: Expected axis has 0 elements, new values have 2 elements"

Maria · Accepted Answer · 2021-01-15 01:49:15Z

1

I am sure that there are way more pythonic solutions, but I could think of this:

from numpy.random import rand
import pandas as pd
IDS = [[1,4],[3,6]]
raws = ['A','B']
df = pd.DataFrame(columns=raws)
print(df)

for ind in IDS:

    for i in ind:
        df = df.append(pd.Series(name = i, dtype='object'))

    for raw in raws:
        for i in ind:
            df[raw][i] = rand(len(ind))
        print(df)

answered Jan 15, 2021 at 1:49

Maria

4076 silver badges15 bronze badges

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

Collectives™ on Stack Overflow

Add multiple data in columns using indexes pandas

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related