我有一个简单的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 的值和is_events_2仅？我尝试了以下操作:
print(df['is_events_1', 'is_events_2'])
但收到错误KeyError: ('is_events_1', 'is_events_2')。
感谢您的帮助。
最佳答案
您需要将其作为列表传递。
print(df[['is_events_1', 'is_events_2']])
关于python - 获取列名称列表的数据框，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/55570017/