如何從熊貓數據框中提取值並將其放入新列表中？

Question

有一個熊貓數據框，我想取出滿足“真相”條件的特定行並將其放入新列表中，我將如何去做呢？ 嘗試設置df.loc [df == 101]的切片操作，但我不斷收到錯誤“ Key Error = 0”

Answer 1

與值比較時，請指定列名稱。 您正在將整個數據幀與df.loc [df == 101]中的101進行比較

df = pd.DataFrame(data=[['Equity(2 [ARNC])', 310],
                     ['Equity(21 [AAME])', 103],
                     ['Equity(24 [AAPL])', 101],
                     ['Equity(25 [ARNC_PR])', 101],
                     ['Equity(31 [ABAX])', 206]], columns=['Equity','Value'])
print df
             Equity           Value
0      Equity(2 [ARNC])        310
1     Equity(21 [AAME])        103
2     Equity(24 [AAPL])        101
3  Equity(25 [ARNC_PR])        101
4     Equity(31 [ABAX])        206


df = df.loc[df['Value']==101]
print df

                 Equity       Value
2     Equity(24 [AAPL])        101
3  Equity(25 [ARNC_PR])        101

從此更新的數據框中，您可以獲得所需的列表。

newlist = df['Equity'].tolist()
print newList
['Equity(24 [AAPL])', 'Equity(25 [ARNC_PR])']