AttributeError:“float”对象在使用if条件时没有属性“isin”

Question

我有两个国家不同年份的人口规模表，如下所示：

   year   pop1          pop2
0   0   1.000000e+08    1.000000e+08
1   1   9.620000e+07    9.970000e+07
2   2   9.254440e+07    9.940090e+07
3   3   8.902771e+07    9.910270e+07
4   4   8.564466e+07    9.880539e+07

该表仅包含前300年的信息

我正在尝试创建一个函数，该函数将显示人口在哪一年减少/增加了2倍、10倍和100倍。在jezrael的帮助下，我创建了以下函数：

def find_year(df,init_pop,multiplier):
    pop_size=init_pop*multiplier
    pop1_values=df['pop1'].unique().tolist()
    pop2_values=df['pop2'].unique().tolist()
    
   s=pd.DataFrame(df.set_index('year').sub(pop_size).abs().idxmin(),columns =[multiplier])
   return(s)

问题是，每个人口的增长率是不同的，如果我检查人口在10年或100年内减少，其中一个人口在300多年后达到这一点，但在这种情况下，我的函数不起作用，并给出了最大值300

例如，如果我运行pop下降0.1和0.01的结果（同一年pop2是错误的）：

find_year(data,100000000,0.1)

>>>     0.1
pop1    59
pop2    300



find_year(data,100000000,0.01)

>>>     0.01
pop1    119
pop2    300

为了修复它，我在函数中创建了条件，检查是否：

pop_size=init_pop*multiplier

位于原始df中的一列中，如果不是，则不会将其添加到结果中。我尝试过这样做：

def find_year(df,init_pop,multiplier):
    pop_size=init_pop*multiplier
    pop1_values=df['pop1'].unique().tolist()
    pop2_values=df['pop2'].unique().tolist()
    
    if pop_size.isin(pop1_values) & pop_size.isin(pop2_values):
        
        s=pd.DataFrame(df.set_index('year').sub(pop_size).abs().idxmin(),columns =[multiplier])
        return(s)
    elif pop_size.isin(pop1_values) & ~pop_size.isin(pop2_values):
        s=pd.DataFrame(df.set_index('year').sub(pop_size).abs().idxmin(),columns =[multiplier])
        s.drop('pop2',axis=0,inplace=True)
        return(s)
    elif pop_size.isin(pop2_values) & ~pop_size.isin(pop1_values):
        
        s=pd.DataFrame(df.set_index('year').sub(pop_size).abs().idxmin(),columns =[multiplier])
        s.drop('pop1',axis=0,inplace=True)
        return(s)
        
        

但当我运行它时，我得到：

AttributeError: 'float' object has no attribute 'isin'

我也试着把数字改成整数，但我还是得到了同样的错误，只是没有“float”，而是int。 我不明白为什么会这样

我的最终目标：添加一个条件，即如果变量“pop_size”不在原始表的pop1或pop2列中，它将从结果df中删除该行（即需要300年）。

编辑：如何生成我的数据：

def growth(p,r,t):
    time=np.arange(0,t+1,1)
    res=[p]
    for t in time:
        p=p*r
        res.append(p)
    return(pd.DataFrame(list(zip(time, res)),columns =['year','pop']))
        
country1=growth(100000000,0.962,300)
country2=growth(100000000,0.997,300)

data=pd.concat([country1, country2['pop']],axis=1)
data.columns=['year','pop1','pop2']
data

Answer 1

您得到错误是因为int和float都没有isin属性。它是在数据帧上定义的。下面是一些可能解决您问题的方法：

def find_year(df, init_pop, multiplier):
    pop_size = init_pop*multiplier
    pop1_values = df['pop1'].unique().tolist()
    pop2_values = df['pop2'].unique().tolist()
    s = pd.DataFrame(df.set_index('year').sub(
        pop_size).abs().idxmin(), columns=[multiplier])

    if pop_size in pop1_values and not pop_size in pop2_values:
        s.drop('pop2', axis=0, inplace=True)
    elif pop_size in pop2_values and not pop_size in pop1_values:
        s.drop('pop1', axis=0, inplace=True)

    return s