3D visualization of linearly non-separable data in 2D

In this recipe, we will learn how we can visualize 2D data that is linearly non-separable in 3D. This is typically used to explain the internal workings of the Support Vector Machines algorithm, which takes lower dimensional data to higher dimensional space so that it can find a plane that separates the data into various clusters neatly.

We will plot both 2D and 3D plots with the same data to visualize it better.

Getting ready

Import the required libraries:

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

How to do it...