遍历某些行

Question

我有一个包含航班始发地和目的地的数据框，如下所示：

Dataframe

我想对一组以某些城市开始或结束的行进行迭代。假设，我只想遍历以肯尼迪国际机场或MEX为起点或终点的航班。

我应该使用iloc吗？

Answer 1

df.loc[df['Origin'].isin(["JFK", "MEX"]) | df['Destination'].isin(["JFK", "MEX"])]

Answer 2

好的，您可以将groupby与agg first和last一起使用

s=df.groupby('Flight').Origin.agg(['first','Last'])
s=s.isin(["JFK", "MEX"]).any(axis=1)
yourdf=df.loc[df.Flight.isin(s[s].index).copy()