python - 从 pandas df 中选择特定列

Question

是否有一种有效的方法可以从 pandas df 中删除 n 个列。例如，如果您不希望超出某一列的任何内容，您可以合并一个函数 df.drop > n

对于下面的 df，我想删除第 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 列，我就必须手动添加或减去列值。如果我经常这样做，就会变得很麻烦。

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 names
print(df)
1   2   3
X   A   C
Y   B   D