下面是一个函数,它接受一个文件并删除列名‘ row _ num’、‘ start _ date’和‘ end _ date’
问题是并非每个文件都有这些列名,因此函数返回一个错误。
我的目标是更改代码,以便在这些列存在时删除它们,但在这些列不存在时不返回错误。
def read_df(file):
df = pd.read_csv(file, na_values=['', ' '])
# Drop useless junk and fill empty values with zero
df = df.drop(['row_num','start_date','end_date','symbol'], axis=1).fillna(0)
df=df[df!=0][:-1].dropna().append(df.iloc[-1])
return df