iris-colored_acatter_matrix.py

from sklearn import datasets
from matplotlib import pyplot as plt

from MachineLearningUtils.DatasetTools import DatasetsTools
from MachineLearningUtils.UsefulPlots import DataPlots


def main():
    # load iris data into DataFrame
    iris_dtst = DatasetsTools(datasets.load_iris)
    iris_df = iris_dtst.data_as_df()
    print("columns: {}".format(list(iris_df)))
    print(iris_dtst.info)
    # iris-colored_acatter_matrix in 3 lines of code
    plotter = DataPlots(df=iris_df, ggplot=True, is_verbose=True)
    fig = plotter.colored_scatter_matrix(df=iris_df, colored_column_name="Target")
    # fig.savefig("iris-colored_acatter_matrix.png")
    plt.show()
    return


if __name__ == '__main__':
    main()