forked from plotly/plotly.py
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.py
More file actions
85 lines (61 loc) · 2.22 KB
/
Copy path__init__.py
File metadata and controls
85 lines (61 loc) · 2.22 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
"""
Built-in datasets for demonstration, educational and test purposes.
"""
def gapminder():
"""
Each row represents a country on a given year.
https://www.gapminder.org/data/
Returns:
A `pandas.DataFrame` with 1704 rows and the following columns: `['country', 'continent', 'year', 'lifeExp', 'pop', 'gdpPercap',
'iso_alpha', 'iso_num']`.
"""
return _get_dataset("gapminder")
def tips():
"""
Each row represents a restaurant bill.
https://vincentarelbundock.github.io/Rdatasets/doc/reshape2/tips.html
Returns:
A `pandas.DataFrame` with 244 rows and the following columns: `['total_bill', 'tip', 'sex', 'smoker', 'day', 'time', 'size']`.
"""
return _get_dataset("tips")
def iris():
"""
Each row represents a flower.
https://en.wikipedia.org/wiki/Iris_flower_data_set
Returns:
A `pandas.DataFrame` with 150 rows and the following columns: `['sepal_length', 'sepal_width', 'petal_length', 'petal_width', 'species',
'species_id']`.
"""
return _get_dataset("iris")
def wind():
"""
Each row represents a level of wind intensity in a cardinal direction, and its frequency.
Returns:
A `pandas.DataFrame` with 128 rows and the following columns: `['direction', 'strength', 'frequency']`.
"""
return _get_dataset("wind")
def election():
"""
Each row represents voting results for an electoral district in the 2013 Montreal mayoral election.
Returns:
A `pandas.DataFrame` with 58 rows and the following columns: `['district', 'Coderre', 'Bergeron', 'Joly', 'total', 'winner', 'result']`.
"""
return _get_dataset("election")
def carshare():
"""
Each row represents the availability of car-sharing services near the centroid of a zone in Montreal.
Returns:
A `pandas.DataFrame` with 249 rows and the following columns: `['centroid_lat', 'centroid_lon', 'car_hours', 'peak_hour']`.
"""
return _get_dataset("carshare")
def _get_dataset(d):
import pandas
import os
return pandas.read_csv(
os.path.join(
os.path.dirname(os.path.dirname(__file__)),
"package_data",
"datasets",
d + ".csv.gz",
)
)