我使用 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 数组中提取第一列。怎么可能?
谢谢
最佳答案
您可以使用ix
并省略 with open('data.csv', 'r') as f:
,因为 read_csv
可以通过文件路径
读取文件。
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]
关于python - 如何获取数组的第一列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35201562/