内置节假日
from prophet import Prophet
import holidays
us_holidays = holidays.US(years = range(2022,2024))
for date, name in sorted(list(us_holidays.items())):
print(f"{date}: {name}")
自定义节假日
from prophet import Prophet
holidays = pd.DataFrame({
'holiday': 'custom_holiday',
'ds': pd.to_datetime(['2023-01-01']),
'lower_window': -50,
'upper_window': 50,
})
model = Prophet(
holidays = holidays
)
内置国家列表
COUNTRIES: RegistryDict = {
"albania": ("Albania", "AL", "ALB"),
"algeria": ("Algeria", "DZ", "DZA"),
"american_samoa": ("AmericanSamoa", "AS", "ASM", "HolidaysAS"),
"andorra": ("Andorra", "AD", "AND"),
"angola": ("Angola", "AO", "AGO"),
"argentina": ("Argentina", "AR", "ARG"),
"armenia": ("Armenia", "AM", "ARM"),
"aruba": ("Aruba", "AW", "ABW"),
"australia": ("Australia", "AU", "AUS"),
"austria": ("Austria", "AT", "AUT"),
"azerbaijan": ("Azerbaijan", "AZ", "AZE"),
"bahamas": ("Bahamas", "BS", "BHS"),
"bahrain": ("Bahrain", "BH", "BAH"),
"bangladesh": ("Bangladesh", "BD", "BGD"),
"barbados": ("Barbados", "BB", "BRB"),
"belarus": ("Belarus", "BY", "BLR"),
"belgium": ("Belgium", "BE", "BEL"),
"belize": ("Belize", "BZ", "BLZ"),
"bolivia": ("Bolivia", "BO", "BOL"),
"bosnia_and_herzegovina": ("BosniaAndHerzegovina", "BA", "BIH"),
"botswana": ("Botswana", "BW", "BWA"),
"brazil": ("Brazil", "BR", "BRA"),
"brunei": ("Brunei", "BN", "BRN"),
"bulgaria": ("Bulgaria", "BG", "BLG"),
"burkina_faso": ("BurkinaFaso", "BF", "BFA"),
"burundi": ("Burundi", "BI", "BDI"),
"cambodia": ("Cambodia", "KH", "KHM"),
"cameroon": ("Cameroon", "CM", "CMR"),
"canada": ("Canada", "CA", "CAN"),
"chad": ("Chad", "TD", "TCD"),
"chile": ("Chile", "CL", "CHL"),
"china": ("China", "CN", "CHN"),
"colombia": ("Colombia", "CO", "COL"),
"costa_rica": ("CostaRica", "CR", "CRI"),
"croatia": ("Croatia", "HR", "HRV"),
"cuba": ("Cuba", "CU", "CUB"),
"curacao": ("Curacao", "CW", "CUW"),
"cyprus": ("Cyprus", "CY", "CYP"),
"czechia": ("Czechia", "CZ", "CZE"),
"denmark": ("Denmark", "DK", "DNK"),
"djibouti": ("Djibouti", "DJ", "DJI"),
"dominican_republic": ("DominicanRepublic", "DO", "DOM"),
"ecuador": ("Ecuador", "EC", "ECU"),
"egypt": ("Egypt", "EG", "EGY"),
"el_salvador": ("ElSalvador", "SV", "SLV"),
"estonia": ("Estonia", "EE", "EST"),
"eswatini": ("Eswatini", "SZ", "SZW", "Swaziland"),
"ethiopia": ("Ethiopia", "ET", "ETH"),
"finland": ("Finland", "FI", "FIN"),
"france": ("France", "FR", "FRA"),
"gabon": ("Gabon", "GA", "GAB"),
"georgia": ("Georgia", "GE", "GEO"),
"germany": ("Germany", "DE", "DEU"),
"greece": ("Greece", "GR", "GRC"),
"guam": ("Guam", "GU", "GUM", "HolidaysGU"),
"guatemala": ("Guatemala", "GT", "GUA"),
"honduras": ("Honduras", "HN", "HND"),
"hongkong": ("HongKong", "HK", "HKG"),
"hungary": ("Hungary", "HU", "HUN"),
"iceland": ("Iceland", "IS", "ISL"),
"india": ("India", "IN", "IND"),
"indonesia": ("Indonesia", "ID", "IDN"),
"iran": ("Iran", "IR", "IRN"),
"ireland": ("Ireland", "IE", "IRL"),
"isle_of_man": ("IsleOfMan", "IM", "IMN"),
"israel": ("Israel", "IL", "ISR"),
"italy": ("Italy", "IT", "ITA"),
"jamaica": ("Jamaica", "JM", "JAM"),
"japan": ("Japan", "JP", "JPN"),
"kazakhstan": ("Kazakhstan", "KZ", "KAZ"),
"kenya": ("Kenya", "KE", "KEN"),
"kyrgyzstan": ("Kyrgyzstan", "KG", "KGZ"),
"laos": ("Laos", "LA", "LAO"),
"latvia": ("Latvia", "LV", "LVA"),
"lesotho": ("Lesotho", "LS", "LSO"),
"liechtenstein": ("Liechtenstein", "LI", "LIE"),
"lithuania": ("Lithuania", "LT", "LTU"),
"luxembourg": ("Luxembourg", "LU", "LUX"),
"madagascar": ("Madagascar", "MG", "MDG"),
"malawi": ("Malawi", "MW", "MWI"),
"malaysia": ("Malaysia", "MY", "MYS"),
"maldives": ("Maldives", "MV", "MDV"),
"malta": ("Malta", "MT", "MLT"),
"marshall_islands": ("MarshallIslands", "MH", "MHL", "HolidaysMH"),
"mexico": ("Mexico", "MX", "MEX"),
"moldova": ("Moldova", "MD", "MDA"),
"monaco": ("Monaco", "MC", "MCO"),
"montenegro": ("Montenegro", "ME", "MNE"),
"morocco": ("Morocco", "MA", "MOR"),
"mozambique": ("Mozambique", "MZ", "MOZ"),
"namibia": ("Namibia", "NA", "NAM"),
"netherlands": ("Netherlands", "NL", "NLD"),
"new_zealand": ("NewZealand", "NZ", "NZL"),
"nicaragua": ("Nicaragua", "NI", "NIC"),
"nigeria": ("Nigeria", "NG", "NGA"),
"north_macedonia": ("NorthMacedonia", "MK", "MKD"),
"northern_mariana_islands": ("NorthernMarianaIslands", "MP", "MNP", "HolidaysMP"),
"norway": ("Norway", "NO", "NOR"),
"pakistan": ("Pakistan", "PK", "PAK"),
"panama": ("Panama", "PA", "PAN"),
"paraguay": ("Paraguay", "PY", "PRY"),
"peru": ("Peru", "PE", "PER"),
"philippines": ("Philippines", "PH", "PHL"),
"poland": ("Poland", "PL", "POL"),
"portugal": ("Portugal", "PT", "PRT"),
"puerto_rico": ("PuertoRico", "PR", "PRI", "HolidaysPR"),
"romania": ("Romania", "RO", "ROU"),
"russia": ("Russia", "RU", "RUS"),
"san_marino": ("SanMarino", "SM", "SMR"),
"saudi_arabia": ("SaudiArabia", "SA", "SAU"),
"serbia": ("Serbia", "RS", "SRB"),
"singapore": ("Singapore", "SG", "SGP"),
"slovakia": ("Slovakia", "SK", "SVK"),
"slovenia": ("Slovenia", "SI", "SVN"),
"south_africa": ("SouthAfrica", "ZA", "ZAF"),
"south_korea": ("SouthKorea", "KR", "KOR", "Korea"),
"spain": ("Spain", "ES", "ESP"),
"sweden": ("Sweden", "SE", "SWE"),
"switzerland": ("Switzerland", "CH", "CHE"),
"taiwan": ("Taiwan", "TW", "TWN"),
"tanzania": ("Tanzania", "TZ", "TZA"),
"thailand": ("Thailand", "TH", "THA"),
"timor_leste": ("TimorLeste", "TL", "TLS"),
"tonga": ("Tonga", "TO", "TON"),
"tunisia": ("Tunisia", "TN", "TUN"),
"turkey": ("Turkey", "TR", "TUR"),
"ukraine": ("Ukraine", "UA", "UKR"),
"united_arab_emirates": ("UnitedArabEmirates", "AE", "ARE"),
"united_kingdom": ("UnitedKingdom", "GB", "GBR", "UK"),
"united_states_minor_outlying_islands": (
"UnitedStatesMinorOutlyingIslands",
"UM",
"UMI",
"HolidaysUM",
),
"united_states_virgin_islands": ("UnitedStatesVirginIslands", "VI", "VIR", "HolidaysVI"),
"united_states": ("UnitedStates", "US", "USA"),
"uruguay": ("Uruguay", "UY", "URY"),
"uzbekistan": ("Uzbekistan", "UZ", "UZB"),
"vanuatu": ("Vanuatu", "VU", "VTU"),
"vatican_city": ("VaticanCity", "VA", "VAT"),
"venezuela": ("Venezuela", "VE", "VEN"),
"vietnam": ("Vietnam", "VN", "VNM"),
"zambia": ("Zambia", "ZM", "ZMB"),
"zimbabwe": ("Zimbabwe", "ZW", "ZWE"),
}