python - 使用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