選擇多列時 Pandas 中的 keyError

Question

在熊貓數據框中選擇多列時出現奇怪的錯誤。 這是代碼：

import pandas as pd
df = pd.read_csv("./Dataset/train.csv", engine='python')
df['eviv1', 'v2a1']

我收到此錯誤消息： KeyError: ('eviv1', 'v2a1')但我在運行時沒有收到錯誤消息

df['eviv1']

df[v2a1]

分別地。 PS：我正在使用 python 3 和 Jupyter Notebook 以及 pandas 版本 0.23.0。

Answer 1

無法以您嘗試的方式對列名（字符串）進行切片。 請試試這個，

df[['eviv1', 'v2a1']]

Answer 2

這里，內括號用於選擇列名，外括號用於選擇數據。 因此，我們使用 2 組方括號：

df[['eviv1', 'v2a1']]