如何從過濾列表中創建新的 Pandas dataframe？

Question

我想知道如何根據一組值創建一個新的數據框。

給定一個值數組：

x1 = ['apple', 'orange', 'banana']

並給出一個數據框：

index, fruit, foo, bar
0, apple, true, false
1, pineapple, false, true
2, orange, true, true
3, banana, true, true
4, pear, true, true

如何創建一個只包含數組 x1 值的新數據框？

Output數據幀：

index, fruit, foo, bar
0, apple, true, false
1, orange, true, true
2, banana, true, true

Answer 1

我認為您需要pandas.Series.isin(values)可用於檢查Series 中的元素是否包含在值中。

>>> import pandas as pd
>>>
>>> df = pd.DataFrame(
...     {"fruit": ["apple", "pineapple", "orange"], "foo": ["true", "false", "true"]}
... )
>>>
>>> df
       fruit    foo
0      apple   true
1  pineapple  false
2     orange   true
>>> df[df.fruit.isin(['apple', 'orange', 'banana'])]
    fruit   foo
0   apple  true
2  orange  true