Python: How to drop a row whose particular column is empty/NaN?

后端 未结 2 551
面向向阳花
面向向阳花 2020-12-01 11:00

I have a csv file. I read it:

import pandas as pd
data = pd.read_csv(\'my_data.csv\', sep=\',\')
data.head()

It has output like:

         


        
2条回答
  •  难免孤独
    2020-12-01 11:32

    Use dropna with parameter subset for specify column for check NaNs:

    data = data.dropna(subset=['sms'])
    print (data)
       id city department   sms  category
    1   2  lhr    revenue  good         1
    

    Another solution with boolean indexing and notnull:

    data = data[data['sms'].notnull()]
    print (data)
       id city department   sms  category
    1   2  lhr    revenue  good         1
    

    Alternative with query:

    print (data.query("sms == sms"))
       id city department   sms  category
    1   2  lhr    revenue  good         1
    

    Timings

    #[300000 rows x 5 columns]
    data = pd.concat([data]*100000).reset_index(drop=True)
    
    In [123]: %timeit (data.dropna(subset=['sms']))
    100 loops, best of 3: 19.5 ms per loop
    
    In [124]: %timeit (data[data['sms'].notnull()])
    100 loops, best of 3: 13.8 ms per loop
    
    In [125]: %timeit (data.query("sms == sms"))
    10 loops, best of 3: 23.6 ms per loop
    

提交回复
热议问题