python - 如何返回二维数组中除一列以外的所有内容？

Question

我的y 变量PRICE 是NumPy 数组中间的一列，它的位置存储在y_idx 中。我希望我的 x 变量是数组中的所有列，y_idx 位置中的列除外。

在 R 中，我会使用负号从这样的选择中删除一列:

# specify x and y variables
y_idx = df.columns.get_loc('PRICE')
y = dataset[:,y_idx]
x = dataset[:,-y_idx]

如何将 x 设置为数组中除 y 设置的列之外的所有列？

Answer 1

我注意到您可能正在使用 pandas dataFrame。在那种情况下:

x= df.drop('PRICE', axis=1)
y = df['PRICE']