如何将处理datetime df的if语句的for循环转换为列表

Question

我试图将下面的for循环with if语句转换成一个列表

# Create dictionary to hold results
    trip_counts = {'AM': 0, 'PM': 0}

# Loop over all trips
for trip in onebike_datetimes:
  # Check to see if the trip starts before noon
  if trip['start'].hour < 12:
    # Increment the counter for before noon
    trip_counts["AM"] += 1
  else:
    # Increment the counter for after noon
    trip_counts["PM"] += 1

我试过了

[trip_counts["AM"]+=1 if trip['start'].hour <12 else trip_counts['PM']+= 1 for trip in onebike_datetimes] 

但我一直收到一个syntaxerror

Answer 1

更清楚的是，保持你的for循环

如果你真的想利用列表理解，你可以这样做：

l = ["AM" if trip["start"].hour < 12 else "PM" for trip in onebike_datetimes]
am_count = l.count("AM")
trip_counts = {"AM": am_count, "PM": len(l) - am_count}

（如果使用此选项，则不需要初始化trip_counts）