Source code for fivpy.datasets

"""
Module with functions to load the data used in examples.
"""

import pandas as pd
import os

[docs]def data_1(): """ Loads the data from the board 4.1. - Credits to Soares, Paula Neto & Souza (2010). The board contains the columns: units: the sample units used columns: the columns selected from the complete data rows: the rows selected from the complete data volume: the volume of the sample unit Returns: data (pandas.DataFrame): The data. Examples -------- >>> data_1() """ filename = os.path.join(os.path.dirname(__file__), 'data/data_1.csv') return pd.read_csv(filename, sep=';', decimal=',')
[docs]def data_2(): """ Loads one of the bult-in datasets with data collected in the Atlantic Forest, Bahia, Brazil. The dataset contains the columns: n_tree: number of the tree on field units: the sample units used cbh: circumference at breast height height: height of the tree Returns: data (pandas.DataFrame): The data. Examples -------- >>> data_2() """ filename = os.path.join(os.path.dirname(__file__), 'data/data_2.csv') return pd.read_csv(filename, sep=';', decimal=',')
[docs]def data_3(): """ Loads one of the bult-in datasets with data collected in the Atlantic Forest, Bahia, Brazil. The dataset contains the columns: n_tree: number of the tree on field units: the sample units used cbh: circumference at breast height height: height of the tree Returns: data (pandas.DataFrame): The data. Examples -------- >>> data_3() """ filename = os.path.join(os.path.dirname(__file__), 'data/data_3.csv') return pd.read_csv(filename, sep=';', decimal=',')
if __name__ == '__main__': print(data_1()) print(data_2()) print(data_3())