到目前为止,我主要是一个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“新手”的身份,我有兴趣知道是否有更好的更简洁的方式。
谢谢
目前没有回答
相关问题 更多 >
编程相关推荐