|
import numpy as np |
|
import pandas as pd |
|
|
|
def main(): |
|
|
|
flights = pd.read_csv("/Users/mwaskom/Desktop/AirPassengers.csv") |
|
flights["year"] = np.floor(flights.time).astype(int) |
|
flights["month"] = np.round((flights.time - flights.year) * 12).astype(int) |
|
|
|
month_number_to_name = {0: "January", |
|
1: "February", |
|
2: "March", |
|
3: "April", |
|
4: "May", |
|
5: "June", |
|
6: "July", |
|
7: "August", |
|
8: "September", |
|
9: "October", |
|
10: "November", |
|
11: "December"} |
|
|
|
flights["month"] = flights.month.map(month_number_to_name) |
|
flights["passengers"] = flights["AirPassengers"] |
|
flights = flights.drop(["AirPassengers", "time", "Unnamed: 0"], axis=1) |
|
flights.to_csv("flights.csv", index=False) |
|
|
|
|
|
if __name__ == "__main__": |
|
main() |
|
|