如何在使用for循环后以数据帧格式打印结果

Question

我正在分析一些数据。我让我的for循环以数组格式打印，但我想以数据帧格式打印结果，因为我正在处理数据帧。

import itertools 

for time, flight in itertools.izip(flight_data["AirTime"], flight_data["UniqueCarrier"]):
    if time > 300:
        print (time, flight)
Outputs:(340, AA)

但是，我希望输出以数据帧格式返回。请协助谢谢

Answer 1

使用boolean indexing：

print (flight_data[flight_data.AirTime > 300])

样品：

flight_data = pd.DataFrame({'AirTime':[340,200,110],
                            'UniqueCarrier':['AA','SS','DD']})

print (flight_data)
   AirTime UniqueCarrier
0      340            AA
1      200            SS
2      110            DD


print (flight_data[flight_data.AirTime > 300])
   AirTime UniqueCarrier
0      340            AA