如何获取数组的第一列

Question

我正在使用csv文件作为我的模型的输入数据。我正在使用pandas数据帧来选择所需的列，如下所示：

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)

然后我想从X_test数组中提取第一列。怎么可能？

谢谢

Answer 1

您可以使用ix并省略with open('data.csv', 'r') as f:，因为read_csv可以按path of file读取文件。

dataframe = pd.read_csv('data.csv')

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

X_train, X_test, y_train, y_test = train_test_split(X, y)

X_test.ix[:,0]