python - 如何从 Pandas Dataframe 中获取标签

Question

我正在尝试实现以下示例代码，但我引用的课程不是很清楚。我知道我正在将数据作为 CSV 文件加载，但我不确定是否dataframe.label应该返回行标签或列标签。此外，我无法弄清楚如何用工作代码替换它。我曾尝试使用iloc, .columns, .to_numpy(),.as_matrix()都无济于事，而我的部分斗争是确定 .label 在某一时刻到底做了什么（或应该做什么）。这是我引用的课程： https ://www.kaggle.com/dansbecker/deep-learning-from-scratch

这是代码：

def data_prep(raw):
    out_y = keras.utils.to_categorical(raw.label, num_classes)

    num_images = raw.shape[0]
    x_as_array = raw.values[:,1:]
    x_shaped_array = x_as_array.reshape(num_images, img_rows, img_cols, 1)
    out_x = x_shaped_array / 255
    return out_x, out_y

train_file = "../input/digit-recognizer/train.csv"
raw_data = pd.read_csv(train_file)

x, y = data_prep(raw_data)

我得到的原始错误是：

AttributeError：“DataFrame”对象没有属性“label”

任何人都可以对此有所了解吗？

标签： pythonpandas