Displaying pair plot in Pandas data frame

后端 未结 5 898
梦如初夏
梦如初夏 2021-02-07 06:19

I am trying to display a pair plot by creating from scatter_matrix in pandas dataframe. This is how the pair plot is created:

# Create dataframe from data in X_t         


        
5条回答
  •  离开以前
    2021-02-07 07:15

    This code worked for me using Python 3.5.2:

    import pandas as pd
    import matplotlib.pyplot as plt
    %matplotlib inline
    
    from sklearn import datasets
    
    iris_dataset = datasets.load_iris()
    X = iris_dataset.data
    Y = iris_dataset.target
    
    iris_dataframe = pd.DataFrame(X, columns=iris_dataset.feature_names)
    
    # Create a scatter matrix from the dataframe, color by y_train
    grr = pd.plotting.scatter_matrix(iris_dataframe, c=Y, figsize=(15, 15), marker='o',
                                     hist_kwds={'bins': 20}, s=60, alpha=.8)
    

    For pandas version < v0.20.0.

    Thanks to michael-szczepaniak for pointing out that this API had been deprecated.

    grr = pd.scatter_matrix(iris_dataframe, c=Y, figsize=(15, 15), marker='o',
                            hist_kwds={'bins': 20}, s=60, alpha=.8)
    

    I just had to remove the cmap=mglearn.cm3 piece, because I was not able to make mglearn work. There is a version mismatch issue with sklearn.

    To not display the image and save it directly to file you can use this method:

    plt.savefig('foo.png')
    

    Also remove

    # %matplotlib inline
    

提交回复
热议问题