如何基于列表选择熊猫行？

Question

这里是需要征求您意见的情况。

我通过使用行获得了如下数据框

df = pd.DataFrame(np.arange(1,400,10).reshape(10, 4), columns=['A', 'B', 'C', 'D'])
df

然后，我有一个列表如下：

array([ 331,  51, 131, 211])

我想获得一个表，其中B列仅包含列表中所述的值。 例如，我手动生成预期的输出，如下所示：

谢谢您的建议。谢谢！

Answer 1

我认为您可以使用熊猫https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.isin.html

的isin函数来做到这一点

df[df["B"].isin(yourarray.tolist())]