如何使用 numpy.ndarray 对熊猫框架进行切片

Posted 2023-03-11

【中文标题】如何使用 numpy.ndarray 对熊猫框架进行切片

【英文标题】：How to use numpy.ndarray to slice a pandas frame

【发布时间】：2021-09-01 15:46:40

【问题描述】：

我有 pandas 框架 F 和 numpy ndarray a，尺寸分别为 m x n 和 m x 1。 a 仅包含 1 和 0，我正在尝试获取一个新的 pandas 框架，它是 F 的行，其中 a 为 1。

我试过F2 = F[a,:]和F2 = F[a.tolist(),:]，都没有成功。请问我该如何解决？

【问题讨论】：

@MustafaAydın 你的回答很好，正是我需要的。我只是（直到现在）还没有解决这个问题。谢谢。

【参考方案1】：

IIUC，boolean indexing 和 loc 应该这样做：

F.loc[a.astype(bool)]