• Return random floats in the half-open int">

    How can I plot a confusion matrix in matplotlib?



    Using imshow method, we can create an image with an input (5, 5) array dimension. After that, we can use the xticks and yticks method to mark the ticks on the axes.

    Steps

    • Return random floats in the half-open interval [5, 5) and interpolation='nearest'.

    • Display data as an image, i.e., on a 2D regular raster, with step 1 data.

    • Get or set the current tick locations and labels of the X-axis, using xticks method.

    • Get or set the current tick locations and labels of the Y-axis, using yticks method.

    • Use plt.show() to show the figure.

    Example

    import matplotlib.pyplot as plt
    import numpy as np
    
    plt.imshow(np.random.random((5, 5)), interpolation='nearest')
    plt.xticks(np.arange(0, 5), ['A', 'B', 'C', 'D', 'E'])
    plt.yticks(np.arange(0, 5), ['1', '2', '3', '4', '5'])
    
    plt.show()

    Output

    Kickstart Your Career

    Get certified by completing the course

    Get Started
    Advertisements