Intellipaat Back

Explore Courses Blog Tutorials Interview Questions
0 votes
2 views
in Data Science by (17.6k points)

I have a dataframe that may or may not have columns that are the same value. For example

    row    A    B

    1      9    0

    2      7    0

    3      5    0

    4      2    0

I'd like to return just

   row    A  

   1      9    

   2      7    

   3      5    

   4      2

Is there a simple way to identify if any of these columns exist and then remove them?

1 Answer

0 votes
by (41.4k points)

I believe this option will be faster than the other answers here as it will traverse the data frame only once for the comparison and short-circuit if a non-unique value is found.

>>> df

   0  1  2

0  1  9  0

1  2  7  0

2  3  7  0

>>> df.loc[:, (df != df.iloc[0]).any()] 

   0  1

0  1  9

1  2  7

2  3  7

If you wish to learn more about Pandas visit this Pandas Tutorial.

31k questions

32.8k answers

501 comments

693 users

Browse Categories

...