<分区>
到目前为止,我主要是 R 用户,现在我正在尝试更好地使用 Python,所以请记住这一点,因为我可能不会以 Python 的方式思考......
无论如何,我想按列位置对 pandas 数据框进行子集化,例如,我会选择前 2 列、第 4 列,然后是最后两列。
我使用的代码如下:
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“新手”的身份,我很想知道是否有更好更简洁的方式。
谢谢,