根据另一列的特定内容打印列内容答案

【问题标题】：Print Column contents, according to particular contents of another column根据另一列的特定内容打印列内容
【发布时间】：2015-06-15 16:28:17
【问题描述】：

我有一个 DataFrame，其中包含trip_id、service_id 等列，例如

trip_id     service_id
1            weekday
2            weekday
3            weekday
4            saturday
5            saturday
6            holiday
7            sunday

我想分别打印所有“工作日”、所有周六和节假日的trip_id。我试过了

join_df.query(join_df['service_id'] == 'weekday')

方式，但它似乎不是正确的方式。我也试过了

print join_df[join_df.service_id =='weekdays']

没用。得到一个空的数据框作为输出。

有人可以帮我解决这个问题吗？谢谢

【问题讨论】：

请提供您的 DataFrame 示例和预期输出。

标签： pandas filter

【解决方案1】：

如果你有以下DataFrame：

       service_id
trip_id           
1          weekday
2          weekday
3          weekday
4         saturday
5         saturday
6          holiday
7           sunday

然后，您可以遍历 service_id 列中的所有唯一值，并使用 boolean indexing 打印出原始 DataFrame 的子集。例如：

for grp in df['service_id'].unique():
    print df[df['service_id'] == grp]

输出：

        service_id
trip_id           
1          weekday
2          weekday
3          weekday

        service_id
trip_id           
4         saturday
5         saturday

        service_id
trip_id           
6          holiday

        service_id
trip_id           
7           sunday

【讨论】：

【解决方案2】：

这看起来像是一个错字（它是工作日而不是工作日），后者或使用 loc 应该可以工作：

In [11]: df[df.service_id == 'weekday']
Out[11]:
   trip_id service_id
0        1    weekday
1        2    weekday
2        3    weekday

In [12]: df.loc[df.service_id == 'weekday']
Out[12]:
   trip_id service_id
0        1    weekday
1        2    weekday
2        3    weekday

您也可以使用查询，但您的语法不太正确：

In [13]: df.query("service_id == 'weekday'")
Out[13]:
   trip_id service_id
0        1    weekday
1        2    weekday
2        3    weekday

【讨论】：