Python 如何获取数组的第一列

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

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_测试数组中提取第一列。怎么可能呢

谢谢

您可以使用open（'data.csv'，r'）作为f:使用和省略

，因为您可以通过文件的
路径读取文件

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]

也许你可以使用
X_-test.ix[：，0]
，你可以用open（'data.csv'，r'）省略
，作为f:
，或者如果
X_-test
是numpy数组-
X-test[：，0]
我认为最好的办法是在你的问题中添加示例-函数的输出是什么
train_-test\u-split（X，y）
？X_-test.ix[：，0]工作得很好。谢谢@jezrael It将X和y分为两类列车和测试数据，用于建模。如果您感兴趣，可以在此处阅读更多内容：