Блог

Главная - Вопросы по программированию - Перебор определенных строк

Перебор определенных строк

#python #pandas

#python #панды

Вопрос:

У меня есть фрейм данных, содержащий начало и пункт назначения рейсов, например:

Фрейм данных

И я хочу выполнить итерацию только по набору строк, которые начинаются или заканчиваются определенными городами. Допустим, я просто хочу выполнить итерацию по рейсам, начинающимся или заканчивающимся в аэропорту JFK или MEX.

Должен ли я использовать iloc?

Комментарии:

1. df.loc[df. Origin.isin([‘JFK’,’MEX’])]

2. Учитывает ли это и назначение? Я хотел бы выполнять итерации не только по рейсам, которые начинаются в каком-то аэропорту, но и по рейсам, которые заканчиваются, и по этим аэропортам

3. Я записал это в ответ

Ответ №1:

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

Ответ №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()