获取列名列表的数据框[重复]

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

有什么方法可以只获取is_events_1 和is_events_2 的data_frame 值吗？我尝试了以下方法：

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

但得到错误KeyError: ('is_events_1', 'is_events_2')。

感谢您的帮助。

Answer 1

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

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