用 pandas 选择前 n 列和后 n 列

Question

我正在尝试使用 pandas 按索引从数据框中选择前 2 列和最后 2 列，并将其保存在同一数据框中。

有没有一种方法可以一步完成？

Answer 1

您可以使用该iloc函数获取列，然后传入索引。

df.iloc[:,[0,1,-1,-2]]

Answer 2

您正在寻找iloc：

df = pd.DataFrame([[1,2,3,4,5], [2,3,4,5,6], [3,4,5,6,7]], columns=['a','b','c','d','e'])

df.iloc[:,:2] # Grabs all rows and first 2 columns
df.iloc[:,-2:] # Grabs all rows and last 2 columns

pd.concat([df.iloc[:,:2],df.iloc[:,-2:]],axis=1) # Puts them together row wise