从 Pandas df 中选择特定列

Question

有没有一种有效的方法drop n的数量columns从一pandas df。例如，如果您不希望某列超出任何内容，可以合并一个函数吗？df.drop > n

对于下面的 df 我想放弃一切过去column 2。我这样做通过。

import pandas as pd

d = ({

    '1' : ['X','Y'],  

    '2' : ['A','B'], 

    '3' : ['C','D'],   

    '4' : ['A','B'], 

    '5' : ['C','D'], 

     })

df = pd.DataFrame(data=d)

df = df.drop(['3','4','5'], axis = 1)

但是，如果有更多或更少的东西，5 columns我将不得不手动添加或减去column values。如果我不断这样做，就会变得很麻烦。

Answer 1

您可以使用df.iloc，df=df.iloc[:,:3] #sliced based on indices. will keeps columns 0 to 2 (indices).(or)df=df.loc[:,:'3'] #sliced based on column namesprint(df)1   2   3X   A   CY   B   D