python-3.x - 我无法在 python 中删除一列

Question

我无法将我的数据集拆分为自变量和因变量

我想将我的数据集拆分为 x 和 y 变量，以便我可以训练它们

df = pd.read_csv('path.csv') 
df.shape
x=df.dropna(["y"])

我尝试过的事情

x=df.dropna(["y"],axis=1)

我想拥有 x 中除 y 列之外的所有值。我得到以下错误

ValueError：对象类型没有名为 y 的轴

标签： python-3.x

Answer 1

您可以使用DataFrame.pop提取列y：

y = df.pop('y')
x = df.copy()

或DataFrame.drop删除列y：

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

样品：

df = pd.DataFrame({
         'a':[4,5,2],
         'b':[7,8,9],
         'c':[1,3,5],
         'y':[5,3,4],
})

y = df.pop('y')
x = df.copy()
print (x)
   a  b  c
0  4  7  1
1  5  8  3
2  2  9  5

print (y)
0    5
1    3
2    4
Name: y, dtype: int64