create a new dataframe from selecting specific rows from existing dataframe python

Question

i have a table in my pandas dataframe. df

id count price
1    2     100
2    7      25
3    3     720
4    7     221
5    8     212
6    2     200

i want to create a new dataframe(df2) from this, selecting rows where count is 2 and price is 100,and count is 7 and price is 221

my output should be df2 =

id count price
1    2     100
4    7     221

i am trying using df[df['count'] == '2' & df['price'] == '100']

but getting error

TypeError: cannot compare a dtyped [object] array with a scalar of type [bool]

It's a classic: df[(df['count'] == '2') & (df['price'] == '100')]. — IanS
– IanS, Commented Nov 30, 2016 at 10:03

jezrael · Accepted Answer · 2024-06-05 08:14:05Z

21

You need to add () because & has higher precedence than ==:

df3 = df[(df['count'] == '2') & (df['price'] == '100')]
print (df3)
  id count price
0  1     2   100

If you need to check multiple values use isin:

df4 = df[(df['count'].isin(['2','7'])) & (df['price'].isin(['100', '221']))]
print (df4)
  id count price
0  1     2   100
3  4     7   221

But if you need to check numeric value of row elements, use:

df3 = df[(df['count'] == 2) & (df['price'] == 100)]
print (df3)

df4 = df[(df['count'].isin([2,7])) & (df['price'].isin([100, 221]))]
print (df4)

edited Jun 5, 2024 at 8:14

answered Nov 30, 2016 at 10:02

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

2 Comments

Shubham R Over a year ago

but how do i join two results in 1 df.? i was thinking of creating two different dataframe for my two rows that i want and then append them? is there any other way to do it?

jezrael Over a year ago

Do you think assign to df3 or df4 ?

Collectives™ on Stack Overflow

create a new dataframe from selecting specific rows from existing dataframe python

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related