Skip to content

A python package for embedding pandas DataFrames as images into pdf and markdown documents

License

Notifications You must be signed in to change notification settings

dexplo/dataframe_image

Repository files navigation

dataframe_image

PyPI - License Python Version

A package to convert pandas DataFrames as images.

Also convert Jupyter Notebooks to PDF and/or Markdown embedding dataframe as image into it.

Exporting individual DataFrames

dataframe_image has the ability to export both normal and styled DataFrames as images from within a Python script. Pass your normal or styled DataFrame to the export function along with a file location to save it as an image.

>>> import dataframe_image as dfi
>>> dfi.export(df_styled, 'df_styled.png')

You may also export directly from the DataFrame or styled DataFrame using the dfi.export and export_png methods, respectively.

>>> import dataframe_image as dfi # you have to import it first to enable this plugin
>>> df.dfi.export('df.png')
>>> df_styled.export_png('df_styled.png')

Here, an example of how exporting a DataFrame would look like in a notebook.

png

Export Jupyter Notebook

When converting Jupyter Notebooks to pdf using nbconvert, pandas DataFrames appear as either raw text or as simple LaTeX tables. The left side of the image below shows this representation.

png

This package was first created to embed DataFrames into pdf and markdown documents as images so that they appear exactly as they do in Jupyter Notebooks, as seen from the right side of the image above. It has since added much more functionality.

Usage

Upon installation, the option DataFrame as Image (PDF or Markdown) will appear in the menu File -> Download as. Clicking this option will open up a new browser tab with a short form to be completed.

png

Installation

Install with either:

  • pip install dataframe_image
  • conda install -c conda-forge dataframe_image

Configuration

table_conversion

When convert dataframe to image, we provide two kind of backend, browser or matplotlib. The default is browser, but you can change it by setting table_conversion parameter to 'matplotlib'.

The major difference between these two backends is that browser backend will render the dataframe as it is in the notebook, while matplotlib backend can work without browser, can export all image format, eg. svg, and will be extremely fast.

But currently matplotlib can only simulate default header and cells style, eg. background_gradient or font-size, any other css properties('width'/'height'/...) or additional element such as set_caption will not work.

dfi.export(df.style.background_gradient(), "df_style.png", table_conversion="matplotlib")

Browser backend

Current we provide 4 difference browser backend liberary: playwright, html2image, selenium and chrome. The default is chrome.

chrome, which means convert image with your local chromium based browser by command line.

html2image is a backup method for chrome, which use html2image.

playwright is a much more stable method, but you have to install playwright first.

selenium is a method that use Firefox driver. Sometimes chrome will make some breaking changes which break methods above, Firefox will be a good backup. Not stable and hard to install. But can be installed in Google Colab.

Other parameters

dfi.export(
    obj: pd.DataFrame,
    filename,
    fontsize=14,
    max_rows=None,
    max_cols=None,
    table_conversion: Literal[
        "chrome", "matplotlib", "html2image", "playwright", "selenium"
    ] = "chrome",
    chrome_path=None,
    dpi=None, # enlarge your image,default is 100,set it larger will get a larger image
    use_mathjax=False, # enable mathjax support, which means you can use latex in your dataframe
)

PDF Conversion - LaTeX vs Chrome Browser

By default, conversion to pdf happens via LaTeX, which you must have pre-installed on your machine. If you do not have the correct LaTeX installation, you'll need to select the Chrome Browser option to make the conversion.

Conversion via Chrome browser is much quicker and will look very different than the LaTeX rendition. The chrome browser version will look nearly the same as it does in your browser, while the LaTeX version looking more like a book/article.

Consult nbconvert's documentation to learn how to get latex installed correctly on your machine.

More features

Below, is a description of other features from dataframe_image:

  • Embeds all images from markdown cells (inline, reference, attachments, and <img> tags) into the pdf
  • Saves the new documents anywhere in your filesystem and correctly link the resources
  • Converts gifs to single-frame png files allowing them to be embedded into the pdf

As a Python Library

dataframe_image can also be used outside of the notebook as a normal Python library. In a separate Python script, import the dataframe_image package and pass the file name of your notebook to the convert function.

>>> import dataframe_image as dfi
>>> dfi.convert('path/to/your_notebook.ipynb',
                to='pdf',
                use='latex',
                center_df=True,
                max_rows=30,
                max_cols=10,
                execute=False,
                save_notebook=False,
                limit=None,
                document_name=None,
                table_conversion='chrome',
                chrome_path=None,
                latex_command=None,
                output_dir=None,
                )

By default, the new file(s) will be saved in the same directory where the notebook resides. Do not run this command within the same notebook that is being converted.

From the Command Line

The command line tool dataframe_image will be available upon installation with the same options as the convert function from above.

dataframe_image --to=pdf "my notebook with dataframes.ipynb" --no-input

Finding Google Chrome

You must have Google Chrome (or Brave) installed in order for dataframe_image to work. The path to Chrome should automatically be found. If Chrome is not in a standard location, set it with the chrome_path parameter.

In Google Colab

A Known Issue: When using dataframe_image with Google Colab, you can not use default Chrome convert method. You can set table_conversion parameter to 'selenium' and it will call Firefox driver to convert the DataFrames to images.

note: you have to install dependencies before use it:

!apt install firefox firefox-geckodriver
!pip install dataframe_image selenium

...
df.dfi.export('df.png', table_conversion='selenium')

Choose your converter

If you do not have Chrome installed or cannot get it to work properly, you can alternatively use matplotlib/selenium to convert the DataFrames to images. Select this option by setting the table_conversion parameter to 'selenium' or 'matplotlib'.

Publish to Medium

Closely related to this package is jupyter_to_medium, which publishes your notebooks directly and quickly as Medium blog posts.

Dependencies

You must have the following Python libraries installed:

About

A python package for embedding pandas DataFrames as images into pdf and markdown documents

Resources

License

Stars

Watchers

Forks

Sponsor this project

Packages

No packages published