函数返回空数据帧

Question

如果我尝试update_figure(2019)，则返回的是空数据框 如果我尝试update_figure('2019')，则返回的是所需的数据帧

有任何解决方法吗？

def update_figure(year):
       df_year = df.loc[df['year'] == year]
       return df_year

Answer 1

您年份列的数据类型为object，即字符串，将其转换为int

df['year'] = df['year'].astype(int)

在执行此操作之前，请检查year列中是否有空值。 我怀疑有空字符串''，如果是，则使用str.replace用所需的整数替换空白空间