获取数据框以获取列名列表

Question

我有一个简单的pandas DataFrame实例，定义如下：

import pandas as pd
timestamps = [1, 2, 3]
is_events_1 = [False, True, True]
is_events_2 = [True, False, True]
is_censored = [False, True, False]
data = dict(is_events_1=is_events_1, is_events_2=is_events_2, is_censored=is_censored)
df = pd.DataFrame(data=data, columns=data.keys(), index=timestamps)

如下所示：

   is_events_1  is_events_2  is_censored
1        False         True        False
2         True        False         True
3         True         True        False

是否有任何方法只能获取data_frame和is_events_1的{​​{1}}值？我尝试了以下操作：

is_events_2

但收到错误 KeyError：（'is_events_1'，'is_events_2'）。

感谢您的帮助。

Answer 1

您需要将其作为列表传递。

print(df[['is_events_1', 'is_events_2']])