数据框子集

Question

我已经有一个数据框（dataframe），并且想把其中的一部分提取到另一个数据框里。

我这样做：

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

不过，我想尝试用整数位置来设置，就像这样：

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

但是我遇到了一个错误。我该怎么做这一部分呢？我看了文档，但没找到明确的答案。

另外，能不能把一个列表作为值传递进去呢？

谢谢。

Answer 1

这个内容在 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