判断DataFrame某一列是否为rangeIndex

我在保存csv的时候,有时候用了Index=False,有的时候没有,就造成读取feature的训练数据的时候,不知道要不要删除第一列。于是可以用这个函数进行判断:

def isDataFrameRangeIndex(df:pd.DataFrame,col):
    return (df.iloc[:, col].values == np.arange(df.shape[0])).all()

你可能感兴趣的:(python代码块)