当前位置:首页 >> 脚本专栏

DataFrame中去除指定列为空的行方法

一次,笔者在处理数据时想去除DataFrame中指定列的值为空的这一行,采用了如下做法,但是怎么都没有成功:

# encoding: utf-8
import pandas as pd
import math
import numpy as np
data = pd.read_csv('mydata.csv')
print len(data)
for i in range(len(data)):
  if (data['导演'][i] == ‘'):
    data = data.drop(i)
data.to_csv('data.csv')

然后发现,要去除DataFrame中指定列的值为空的这一行,可以在循环中使用numpy或者math中的isnan()判断方法:

# encoding: utf-8
import pandas as pd
import math
import numpy as np
data = pd.read_csv('mydata.csv')
print len(data)
for i in range(len(data)):
  if np.isnan(data['导演'][i]):
    data = data.drop(i)
data.to_csv('data.csv')

以上这篇DataFrame中去除指定列为空的行方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。