Source code for featuretools.demo.weather

import pandas as pd

import featuretools as ft


[docs]def load_weather(nrows=None, return_single_table=False): """ Load the Australian daily-min-temperatures weather dataset. Args: nrows (int): Passed to nrows in ``pd.read_csv``. return_single_table (bool): Exit the function early and return a dataframe. """ filename = "daily-min-temperatures.csv" print("Downloading data ...") url = "https://api.featurelabs.com/datasets/{}?library=featuretools&version={}".format( filename, ft.__version__ ) data = pd.read_csv(url, index_col=None, nrows=nrows) if return_single_table: return data es = make_es(data) return es
def make_es(data): es = ft.EntitySet("Weather Data") es.add_dataframe( data, dataframe_name="temperatures", index="id", make_index=True, time_index="Date", ) return es