如何在数据集的特定列中提取元素?

什么 @w-m heir answer 是正确的,您当前正在分配所有行(第一行) : )所有列,从0开始到列1,不包括上限,到 x (与 :1 )所有行(还是第一行) : y (与 :2 ).

x = data[:, 0]
y = data[:, 1]

是正确实现这一点的一种方法,但更好、更简洁的方法是使用元组解包:

x, y = data.T

这就使(`T)数据互换,即交换两个维度,然后第一个维度的长度为2。如果实际数据的列数大于此数,则可以使用:

x, y, *rest = data.T

rest This syntax was introduced in Python 3.0 .

documentation

使第二列的形状与 x ,使用 y=data[:, 1:2] .

注意:您正在使用此索引创建二维数组(形状为(len(data),1))。如果需要1d数组,只需对第二项使用整数,而不是切片:

x = data[:, 0]
y = data[:, 1]