How to get first column of an array
<p>I'm using a csv file as input data for my model. I'm using pandas dataframe to choose desired column of it as follows:</p>

<pre><code>with open('data.csv', 'r') as f:
     dataframe = pd.read_csv(f)

X = dataframe.iloc[:,(0,1)]
y = dataframe.iloc[:,(2)]

X_train, X_test, y_train, y_test = train_test_split(X, y)
</code></pre>

<p>Then I'd like to extract first column from X_test array. How is it possible?</p>

<p>Thanks </p>