ValueError: Masked arrays must be 1-D

The following code was written in python to generate a scatter plot of the test data after splitting it into test and train data. The data being read from a sample file was converted to matrix to be able to use the data for the the linear regression algorithm.

import numpy as np import matplotlib.pyplot as plt import pandas as pd data=pd.read_csv('data_1.csv',encoding='utf-8') print data.head() X_data=np.matrix(data['X1']) Y_data=np.matrix(data['Y1']) X_train,X_test,Y_train,Y_test=train_test_split(np.transpose(X_data),\ np.transpose(Y_data), test_size=0.3) plt.scatter(X_test, Y_test, color='black')

But using the same data for the scatter plot generated the following error.

ValueError: Masked arrays must be 1-D

This is because we are passing X_test and Y_test and matrices to scatter plot but which actually are one dimensional arrays. Thus to get around the problem we can cast the data into an numpy array.

plt.scatter(np.array(X_test), np.array(Y_test), color='black')

The above modification should stop the error from being thrown. The modified code will be.

import numpy as np import matplotlib.pyplot as plt import pandas as pd data=pd.read_csv('data_1.csv',encoding='utf-8') print data.head() X_data=np.matrix(data['X1']) Y_data=np.matrix(data['Y1']) X_train,X_test,Y_train,Y_test=train_test_split(np.transpose(X_data),\ np.transpose(Y_data), test_size=0.3) plt.scatter(np.array(X_test), np.array(Y_test), color='black')