How to create a new dataframe from existing dataframe with certain condition - python

Question

I have a dataframe looks like below.

   text      country    language
-----------------------------------
  football     US         Eng
  baseball     JP         Jpn
  swimming     UK         Eng
  running      FR         Fra
  rugby        NZ         Eng
  Hockey       NL         Dut

In python, I want to extract rows which contain strings 'ball' and 'ing' in the column 'text' and make a new dataframe with those rows like below.

   text      country    language
-----------------------------------
  football     US         Eng
  baseball     JP         Jpn
  swimming     UK         Eng
  running      FR         Fra

Ricardo · Accepted Answer · 2017-11-20 14:11:59Z

3

Using pandas, you can slice using multiple conditions, just watch out for tricky parenthesis on the syntax.

df = pd.DataFrame({'A': ['foo', 'bar', 'fooing', 'barball'],
                   'B': [1, 2, 3, 4]})

df_slice = df[(df.A.str.contains('ing')) | (df.A.str.contains('ball'))]

That should yield

df_slice
A        B
fooing   3
barball  4

If your goal is to slice in words that end in ing or ball, use endswith() instead of contains() in the conditions. Hope it helps!

answered Nov 20, 2017 at 14:11

Ricardo

3352 silver badges8 bronze badges

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

3 Comments

Edward M. Over a year ago

Thanks! Can I ask you one more? What if there are multiple conditions (let's say more than two)? Is it possible to do it like condition = [~,~,~,~], df.A.str.contains(condition)?

Ricardo Over a year ago

I am not sure I understand your question. df.A.str.contains() takes only a str as argument, not a condition. You can chain conditions with & and | and a few parenthesis on the slice, of course. The goal is to create a series with True or False elements and feed it to the dataframe, this will create the requested slice.

Edward M. Over a year ago

I see. Thanks a lot!

Collectives™ on Stack Overflow

How to create a new dataframe from existing dataframe with certain condition - python

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related