熊猫选择单独的列[重复]

Question

给定以下数据框：

import pandas as pd
df = pd.DataFrame({'a':[1,2,3],'b':[4,5,6],'c':[1,2,3],'d':[4,5,6]})

df
   a  b  c  d
0  1  4  1  4
1  2  5  2  5
2  3  6  3  6

我只想通过列索引位置选择列 a、c 和 d。 我尝试了以下方法，但没有成功：

df[df.columns[0],df.columns[-2:]]

我可能可以通过常规 Python 解决，但想知道 pandas 中是否有捷径。

Answer 1

使用索引append

df.loc[:,df.columns[[0]].append(df.columns[-2:])]
Out[1363]: 
   a  c  d
0  1  1  4
1  2  2  5
2  3  3  6

或者也许只是使用.iloc

df.iloc[:,[0,2,3]] 

嗯，也许np.r_

df.iloc[:,np.r_[0,-2:0]]
Out[1371]: 
   a  c  d
0  1  1  4
1  2  2  5
2  3  3  6