In-depth EDA (target analysis, comparison, feature analysis, correlation) in two lines of code!
Sweetviz is an open-source Python library that generates beautiful, high-density visualizations to kickstart EDA (Exploratory Data Analysis) with just two lines of code. Output is a fully self-contained HTML application.
The system is built around quickly visualizing target values and comparing datasets. Its goal is to help quick analysis of target characteristics, training vs testing data, and other such data characterization tasks.
Usage and parameters are described below, you can also find an article describing its features in depth and see examples in action HERE.
Sweetviz development is still ongoing! Please let me know if you run into any data, compatibility or install issues! Thank you for reporting any BUGS in the issue tracking system here, and I welcome your feedback and questions on usage/features in the brand-new GitHub "Discussions" tab right here!.
(see below for docs on these features)
Some people have experienced mixed results behavior upgrading through
pip. To update to the latest from an existing install, it is recommended to
pip uninstall sweetviz first, then simply install.
Sweetviz currently supports Python 3.6+ and Pandas 0.25.3+. Reports are output using the base "os" module, so custom environments such as Google Colab which require custom file operations are not yet supported, although I am looking into a solution.
The best way to install sweetviz (other than from source) is to use pip:
pip install sweetviz
In some rare cases, users have reported errors such as
ModuleNotFoundError: No module named 'sweetviz' and
AttributeError: module 'sweetviz' has no attribute 'analyze'.
In those cases, we suggest the following:
sweetviz.py, as that interferes with the library itself. Delete or rename that script (and any associated
.pycfiles), and try again.
pip uninstall sweetviz, then reinstalling
Creating a report is a quick 2-line process:
DataframeReportobject using one of:
show_xxx()function to render the report. You can now use either html or notebook report options, as well as scaling: (more info on these options below)
There are 3 main functions for creating reports:
To analyze a single dataframe, simply use the
analyze(...) function, then the
import sweetviz as sv my_report = sv.analyze(my_dataframe) my_report.show_html() # Default arguments will generate to "SWEETVIZ_REPORT.html"
analyze() function can take multiple other arguments:
analyze(source: Union[pd.DataFrame, Tuple[pd.DataFrame, str]], target_feat: str = None, feat_cfg: FeatureConfig = None, pairwise_analysis: str = 'auto'):
force_text. The "force_" arguments override the built-in type detection. They can be constructed as follows:
feature_config = sv.FeatureConfig(skip="PassengerId", force_text=["Age"])
="off") since processing that many features would take a long time. This parameter also covers the generation of the association graphs (based on Drazen Zaric's concept):
To compare two data sets, simply use the
compare() function. Its parameters are the same as
analyze(), except with an inserted second parameter to cover the comparison dataframe. It is recommended to use the [dataframe, "name"] format of parameters to better differentiate between the base and compared dataframes. (e.g.
[my_df, "Train"] vs
my_report = sv.compare([my_dataframe, "Training Data"], [test_df, "Test Data"], "Survived", feature_config)
Another way to get great insights is to use the comparison functionality to split your dataset into 2 sub-populations.
Support for this is built in through the
compare_intra() function. This function takes a boolean series as one of the arguments, as well as an explicit "name" tuple for naming the (true, false) resulting datasets. Note that internally, this creates 2 separate dataframes to represent each resulting group. As such, it is more of a shorthand function of doing such processing manually.
my_report = sv.compare_intra(my_dataframe, my_dataframe["Sex"] == "male", ["Male", "Female"], feature_config)
Once you have created your report object (e.g.
my_report in the examples above), simply pass it into one of the two `show' functions:
show_html( filepath='SWEETVIZ_REPORT.html', open_browser=True, layout='widescreen', scale=None)
show_html(...) will create and save an HTML report at the given file path. There are options for:
'vertical'. The widescreen layout displays details on the right side of the screen, as the mouse goes over each feature. The new (as of 2.0) vertical layout is more compact horizontally and enables expanding each detail area upon clicking.
None) to scale the entire report. This is very useful to fit reports to any output.
show_notebook( w=None, h=None, scale=None, layout='widescreen', filepath=None)
show_notebook(...) is new as of 2.0 and will embed an IFRAME element showing the report right inside a notebook (e.g. Jupyter, Google Colab, etc.).
Note that since notebooks are generally a more constrained visual environment, it is probably a good idea to use custom width/height/scale values (
scale) and even set custom default values in an INI override (see below). The options are:
scalefor the report itself). Can be as a percentage string (
w="100%") or number of pixels (
h=700) or "Full" to stretch the window to be as tall as all the features (
The package contains an INI file for configuration. You can override any setting by providing your own then calling this before creating a report:
IMPORTANT #1: it is best to load overrides before any other command, as many of the INI options are used in the report generation.
IMPORTANT #2: always put the header line (e.g.
[General]) before a set of values in your override INI file, otherwise your settings will be ignored. See examples below.
You can look into the file
sweetviz_defaults.ini for what can be overriden (warning: much of it is a work in progress and not well documented), but the most useful overrides are as follows.
Override any of these (by putting them in your own INI, again do not forget the header), to avoid having to set them every time you do a "show" command:
Important: note the double '%' if specifying a percentage
[Output_Defaults] html_layout = widescreen html_scale = 1.0 notebook_layout = vertical notebook_scale = 0.9 notebook_width = 100%% notebook_height = 700
[General] use_cjk_font = 1
Will switch the font in the graphs to use a CJK-compatible font. Although this font is not as compact, it will get rid of any warnings and "unknown character" symbols for these languages.
[Layout] show_logo = 0
Will remove the Sweetviz logo from the top of the page.
A major source of insight and unique feature of Sweetviz' associations graph and analysis is that it unifies in a single graph (and detail views):
Squares represent categorical-featured-related variables and circles represent numerical-numerical correlations. Note that the trivial diagonal is left empty, for clarity.
IMPORTANT: categorical-categorical associations (provided by the SQUARES showing the uncertainty coefficient) are ASSYMMETRICAL, meaning that each row represents how much the row title (on the left) gives information on each column. For example, "Sex", "Pclass" and "Fare" are the elements that give the most information on "Survived".
For the Titanic dataset, this information is rather symmetrical but it is not always the case!
Correlations are also displayed in the detail section of each feature, with the target value highlighted when applicable. e.g.:
Finally, it is worth noting these correlation/association methods shouldn’t be taken as gospel as they make some assumptions on the underlying distribution of data and relationships. However they can be a very useful starting point.
As of 2.1, Sweetviz now fully integrates Comet.ml. This means Sweetviz will automatically log any reports generated using
show_notebook() to your workspace, as long as your API key is set up correctly in your environment.
Additionally, you can also use the new function
report.log_comet(experiment_object) to explicitly upload a report for a given experiment to your workspace.
You can customize how the Sweetviz report looks in your Comet workspace by overriding the
[comet_ml_defaults] section of configuration file. See above for more information on using the INI override.
You can choose to use either the
widescreen (horizontal) or
vertical layouts, as well as set your preferred scale, by putting the following in your override INI file:
[comet_ml_defaults] html_layout = vertical html_scale = 0.85
Please see the "Installation issues & fixes" section at the top of this document
See section above regarding CJK characters support. If you find the need for additional character types, definitely post a request in the issue tracking system.
Development is ongoing so absolutely feel free to report any issues and/or suggestions in the issue tracking system here or in our forum (you should be able to log in with your Github account!)
This is my first open-source project! I built it to be the most useful tool possible and help as many people as possible with their data science work. If it is useful to you, your contribution is more than welcome and can take many forms:
A STAR here on GitHub, and a Twitter or Instagram post are the easiest contribution and can potentially help grow this project tremendously! If you find this project useful, these quick actions from you would mean a lot and could go a long way.
Kaggle notebooks/posts, Medium articles, YouTube video tutorials and other content take more time but will help all the more!
I expect there to be many quirks once the project is used by more and more people with a variety of new (& "unclean") data. If you found a bug, please open a new issue here.
To make Sweetviz as useful as possible we need to hear what you would like it to do, or what it could do better! Head on to our Discourse server and post your suggestions there; no login required!.
I definitely welcome the help I can get on this project, simply get in touch on the issue tracker and/or our Discourse forum.
Please note that after a hectic development period, the code itself right now needs a bit of cleanup. :)
I want Sweetviz to be a hub of the best of what's out there, a way to get the most valuable information and visualization, without reinventing the wheel.
As such, I want to point some of those great resources that were inspiring and integrated into Sweetviz:
And of course, very special thanks to everyone who have contributed on Github, through reports, feedback and commits!