DataFrame子集

Question

我已经有了一个数据框，并将其中的一些数据集分配给另一个数据帧。

我这样做：

D = njm[['svntygene', 'intgr', 'lowgr', 'higr', 'lumA', 'lumB', 'wndres', 'nlbrst',        'Erneg', 'basallike']]

我想尝试按整数位置设置它，如下所示：

D = njm.iloc[1:, 2:, 3:, 7:]

但是我收到了一个错误。我该怎么做这部分？阅读文档，但找不到明确的答案。

此外，是否可以将列表作为值传递给它？

感谢。

Answer 1

documentation的iloc部分介绍了这一点：您可以传递包含所需索引的列表。

>>> df = pd.DataFrame(np.random.random((5,5)),columns=list("ABCDE"))
>>> df
          A         B         C         D         E
0  0.605594  0.229728  0.390391  0.754185  0.516801
1  0.384228  0.106261  0.457507  0.833473  0.786098
2  0.364943  0.664588  0.330835  0.846941  0.229110
3  0.025799  0.681206  0.235821  0.418825  0.878566
4  0.811800  0.761962  0.883281  0.932983  0.665609
>>> df.iloc[:,[1,2,4]]
          B         C         E
0  0.229728  0.390391  0.516801
1  0.106261  0.457507  0.786098
2  0.664588  0.330835  0.229110
3  0.681206  0.235821  0.878566
4  0.761962  0.883281  0.665609