當列名是整數時索引熊貓數據框

Question

我似乎無法使用 loc 命令使用整數列名對數據進行子集化

# 6*4 data set with column names as x,y,8,9

df = pd.DataFrame(np.random.randint(0,10,(6,4)),
                  index=('a','b','c','1','2','3'),
                  columns=['x','y', 8, 9])

df2 = df.loc[:,:'x']
df3 = df.loc[:,:'8']

df2 工作，但 df3 拋出錯誤。

Answer 1

您可以執行以下任一操作：

df3 = df.loc[:,8]

只獲得第8列

或者：

df3 = df.loc[:,df.columns[:list(df.columns).index(8)+1]]

獲取第8列之前的所有列（包括 - 刪除+1以獲得獨占）。