gpt4 book ai didi

python - 如何获取数组的第一列

转载 作者:行者123 更新时间:2023-12-01 04:09:04 32 4
gpt4 key购买 nike

我使用 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/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com