Merge rows dataframe based on two columns - Python

Question

I have a data frame df that has the following information:

|   Project | class   | x     |y    | z   |
| ---       | ---     | ---   | --- | --- | 
| Project_A | c.java  |a1     | a2  |     |
| Project_A | c.java  |       |     | a3  |
| Project_b | t.java  |b1     |b2   |     |

I need to combine these lines if the values of project and Class are equal among them.

Based on the previous example, the aspected output in this case have be:

| Project   | class  | x     |y    | z   |
| ---       | ---    | ---   | --- | --- |
| Project_A | c.java |a1     |a2   | a3  |
| Project_b | t.java |b1     |b2   |     |

it is also important to note that the way the dataset was constructed, there is no risk of rewriting values; so, in other words, you will never have such a situation:

| Project   | class  | x   |y   | z   |
| ---       | ---    | --- | ---| --- |
| Project_A | c.java |a1   | a2 |     |
| Project_A | c.java |a_x  | a_y|a_z  |
| Project_b | t.java |b1   |b2  |     |

How can do it?

Corralien · Accepted Answer · 2022-07-27 20:05:11Z

2

This will groupby the project and class and then find the first value of the grouping. I'm not sure if your data will allow for something like this since if there is another example of data in the column for the project/class combination it might mess up some of your data

df.groupby(['Project', 'class'], as_index=False).agg('first')

edited Jul 27, 2022 at 20:05

Corralien

121k8 gold badges44 silver badges69 bronze badges

answered Jul 27, 2022 at 19:59

ArchAngelPwn

3,0461 gold badge6 silver badges17 bronze badges

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

Comments

NirF · Accepted Answer · 2022-07-27 20:02:43Z

1

You can use groupby() on both columns and sum the others:

X = pd.DataFrame({
        'Project':["Project_A","Project_A",'Project_b'],
        'class':["c.java","c.java","t.java"],
        'x':["a1",None,"b1"],
        'y':["a2",None,"b2"],
        'z':[None,"a3",None]})

Y= X.groupby(['Project','class']).sum()
print(Y)

Output:

                   x   y   z
Project   class
Project_A c.java  a1  a2  a3
Project_b t.java  b1  b2   0

answered Jul 27, 2022 at 20:02

NirF

2791 silver badge11 bronze badges

Collectives™ on Stack Overflow

Merge rows dataframe based on two columns - Python

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