https://github.com/koaning/drawdata

Draw datasets from within Jupyter.
https://github.com/koaning/drawdata

data drawdata jupyter

Last synced: 15 days ago
JSON representation

Draw datasets from within Jupyter.

• Host: GitHub
• URL: https://github.com/koaning/drawdata
• Owner: koaning
• License: mit
• Created: 2021-04-04T20:28:10.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-07-15T09:53:47.000Z (4 months ago)
• Last Synced: 2024-10-03T10:34:24.954Z (about 1 month ago)
• Topics: data, drawdata, jupyter
• Language: JavaScript
• Homepage: https://calmcode.io/labs/drawdata.html
• Size: 4.93 MB
• Stars: 763
• Watchers: 8
• Forks: 78
• Open Issues: 2
• Metadata Files:
  • Readme: readme.md
  • License: LICENSE

Awesome Lists containing this project

README

        


### drawdata 

> "Just draw some data and get on with your day."

This small Python library contains Jupyter widgets that allow you to draw a dataset in a Jupyter

notebook. This should be very useful when teaching machine learning algorithms.

![](imgs/widget.gif)

The project uses [anywidget](https://anywidget.dev/) under the hood so our tools should work in Jupyter, VSCode and Colab. That also means that you get a proper widget that can interact with [ipywidgets](https://ipywidgets.readthedocs.io/en/stable/) natively. [Here](https://www.youtube.com/watch?v=STPv0jSAQEk) is an example where updating a drawing triggers a new scikit-learn model to train ([code](https://github.com/probabl-ai/youtube-appendix/blob/main/04-drawing-data/notebook.ipynb)).

![](imgs/update.gif)

You can really get creative with this in a notebook, so feel free to give it a spin!

#### Installation 

Installation occurs via pip. 

```

python -m pip install drawdata

```

To read the data, `polars` is useful, but this library also suppots `pandas`:

```

python -m pip install pandas polars

```

#### Usage

You can load the scatter widget to start drawing immediately. 

```python

from drawdata import ScatterWidget

widget = ScatterWidget()

widget

```

If you want to use the dataset that you've just drawn you can do so via: 

```python

# Get the drawn data as a list of dictionaries

widget.data

# Get the drawn data as a dataframe

widget.data_as_pandas

widget.data_as_polars

```

If you're eager to do scikit-learn stuff with your drawn data you may appreciate this property instead:

```

X, y = widget.data_as_X_y

```

The assumption for this property is that if you've used multiple colors that you're interested in doing classification and if you've only drawn one color you're interested in regression. In the case of regression `y` will refer to the y-axis.

#### Shoutout 

This project was originally part of my work over at [calmcode labs](https://calmcode.io/labs/drawdata) but my employer [probabl](https://probabl.ai) has

been very supportive and has allowed me to work on this project during

my working hours. This was super cool and I wanted to make sure I recognise them for it.



    

        

    

    

        

    

    









### Old Features 

The original implementation of our widget would use an iframe to load a site in order

to be able to draw from a Jupyter notebook. This works, but requires more manual effort, only works with `pandas` via the clipboard feature and needs an internet connection. Here's what that widget looks like:

![](imgs/gif.gif)

It will be kept around, but the way forward for this library is to build on top of anywidget.

#### Old Feature Usage 

When you run this from jupyter, you should load in an iframe.

```python

from drawdata import draw_scatter

draw_scatter()

```

Once you're done drawing you can copy the data to the clipboard. After this you can use pandas to read the clipboard to get your drawn data into a dataframe.

```python

import pandas as pd 

pd.read_clipboard(sep=",")

```