返回引用PD Dataframe的方法中的列表

时间:2017-03-16 00:45:56

标签: python pandas

引用pandas DF时有没有办法返回列表或元组? get_df()是一个带有几百个浮点值的pandas列。下面的代码要求返回大于6000且小于7000的值。我可以将列表返回给我的方法吗? (我知道我可以打印这个,但这不是我想要做的)

def mass_needed(numb_one, numb_two):
    for i in get_df():
        if i > numb_one and i < numb_two:
            return(i)
print(mass_needed(6000, 7000))

我想要完成的是我希望能够调用mass_needed()并获取一个我可以打印或操作的列表值,就像普通列表一样。

1 个答案:

答案 0 :(得分:0)

如果有人关心,我想通了。不得不在迭代过程中附加值。

def mass_needed(numb_one, numb_two):
    li = []
    for i in get_df():
        if i > numb_one and i < numb_two:
            li.append(i)
    return li        


x = pd.DataFrame(mass_needed(6000, 7000))
print(x)