我正在处理钻石数据集的样本。我需要编写一个函数,该函数应该输出price列的条目,为此,carat列的条目大于数字1.1且小于1.4。
以下是示例数据:
def exercise1(df,rowinds,colinds):
df_1 = DataFrame(df)
df_2 = df_1.iloc[rowinds, colinds]
return df_2
test_df = exercise1(diamonds,np.arange(500),np.arrange(10))
test_df
到目前为止,这是我的工作:
def exercise2(df,col,output_label,val1,val2):
df1 = DataFrame(df)
df2 = DataFrame(df1,
columns=[col])
df3 = df2.rename(columns = {col:output_label})
df4 = df3[df3 > val1]
df5 = df4[df4 < val2]
df6 = df5.dropna()
return df6
以下测试应返回一系列编号为172和376的钻石,但是我的钻石正在返回编号为172和136的克拉。
test_df = exercise1(diamonds,np.arange(500),np.arange(10))
exercise2(test_df,test_df.carat,'price',1.1,1.4)
我不知道我要去哪里。