如何在数据框内满足布尔条件

时间:2019-10-19 17:29:34

标签: python pandas numpy dataframe boolean

我正在处理钻石数据集的样本。我需要编写一个函数,该函数应该输出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)

我不知道我要去哪里。

0 个答案:

没有答案