按索引(列)numb在pandas数据框中选择列

2024-05-16 04:10:38 发布

您现在位置:Python中文网/ 问答频道 /正文

到目前为止,我主要是一个R用户,我现在正试图用Python做得更好,所以请记住这一点,因为我可能没有用Python的方式思考。。。

在任何情况下,这里,我想按列位置对pandas数据框进行子集,在这里我将选择例如前两列、第四列和最后两列。

我使用的代码如下:

df01 = pd.DataFrame(np.random.randint(low=0, high=10, size=(10, 10)),
                columns=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i','j'])
df01.iloc[:,list(range(0,2)) + list([3]) + list(range(-3,-1))]

我是通过创建3个包含我想要的列的列表来进行子集设置的,但是我认为必须有更好的方法来实现这一点,因为这在我看来太麻烦了。在R中,我可以做一个简单的:

df01[c(1:2,4,9:10)]

再说一次,这可能正是它的样子,但鉴于我作为一个python“新手”的身份,我有兴趣知道是否有更好的更简洁的方式。

谢谢


Tags: 数据代码用户dataframepandasnp方式情况