如何从熊猫数据框中提取值并将其放入新列表中？

Question

有一个熊猫数据框，我想取出满足“真相”条件的特定行，并将它们放入新列表中，我将如何去做呢？ 尝试设置df.loc [df == 101]的切片操作，但我不断收到错误“ Key Error = 0”

Answer 1

请在与值进行比较时指定列名。您正在将整个数据帧与df.loc [df == 101]中的101进行比较

C

从此更新的数据框中，您可以获得所需的列表。

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