关于#python#的问题:关于loc查询结果的问题

import numpy as np
import pandas as pd

list1 = [[5,2,0,4],[5,3,4,5],[3,6,1,9],[5,1,0,8]]
df=pd.DataFrame(list1,index=list('1234'),columns=list('ABCD'))
print(df)

list_a=df.loc[df['A']==5,['B']].values
list_b=[2,3,1]
print(list_a,type(list_a))
print(list_b,type(list_b))

以上代码,查询到了list_a的数据,但希望通过dataframe查询出list_b的数据形式,不知道哪位指点一下?

import numpy as np
import pandas as pd
 
list1 = [[5,2,0,4],[5,3,4,5],[3,6,1,9],[5,1,0,8]]
df = pd.DataFrame(list1,index=list('1234'), columns=list('ABCD'))

 
list_a = df.loc[df['A']==5,'B'].values
print(list(list_a))

有帮助的话,请点采纳该答案~

感谢

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^