如何使用 numpy.ndarray 分割 pandas 帧
How to use numpy.ndarray to slice a pandas frame
我有 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(),:]
,都没有成功。请问我该如何解决?
IIUC,boolean indexing 和 loc
应该做:
F.loc[a.astype(bool)]
我有 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(),:]
,都没有成功。请问我该如何解决?
IIUC,boolean indexing 和 loc
应该做:
F.loc[a.astype(bool)]