Converting a data frame - Python

Question

My data are in a data frame. It looks like this.

I am trying to convert the data frame into the following format:

Currently I'm doing this conversion using the following Python codes, which are tedious.

def f(row):
    if row['d1'] == 1:
        return 1
    if row['d1'] == 2:
        return ''
    if row['d1'] == 3:
        return ''
    if row['d1'] == 4:
        return ''
    else:
        return ''
    return np.nan

df['t1']=df.apply(f, axis=1)

Any help would be greatly appreciated.

Hi Kevin, Could you please explain what df['t1']=df.apply(f, axis=1) does? — Abhishek Sharma
– Abhishek Sharma, Commented Dec 14, 2015 at 19:45
It read the value in d1 column and insert "1" into t1 column (if the value in d1 is 1). My code is poor. That's why I was looking for better codes. — kevin
– kevin, Commented Dec 14, 2015 at 19:49

Anton Protopopov · Accepted Answer · 2015-12-14 19:41:37Z

5

You could apply following function:

def func(x):
    x.iloc[x.iloc[0]] = 1
    return x

In [66]: df
Out[66]: 
   d1  t1  t2  t3  t4  t5
0   5   0   0   0   0   0
1   1   0   0   0   0   0
2   3   0   0   0   0   0
3   4   0   0   0   0   0
4   1   0   0   0   0   0
5   2   0   0   0   0   0

In [67]: df.apply(func, axis=1)
Out[67]: 
   d1  t1  t2  t3  t4  t5
0   5   0   0   0   0   1
1   1   1   0   0   0   0
2   3   0   0   1   0   0
3   4   0   0   0   1   0
4   1   1   0   0   0   0
5   2   0   1   0   0   0

answered Dec 14, 2015 at 19:41

Anton Protopopov

31.9k13 gold badges93 silver badges96 bronze badges

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

2 Comments

kevin Over a year ago

You're awesome! Thanks a lot!

Anton Protopopov Over a year ago

I'm glad I was able to help

Collectives™ on Stack Overflow

Converting a data frame - Python

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related