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

Question

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的数据形式，不知道哪位指点一下？

Answer 1

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))

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