将列名称分配给csv数据集

Question

我目前正在处理包含以下数据的数据集：

paper_id, word_attributes, class_label

现在总共有3700个word_attributes列表示一个二进制值。 python中是否有一种方法可以用来分配列标题？ 谢谢。

Answer 1

您也许可以使用以下命令读取csv文件：

a = np.genfromtxt(filename, delimiter=',', dtype=None, names=True)

它将创建一个numpy.recarray ，其中的每一列都可以通过键来调用，例如a['paper_id'] 。 当dtype=None ，“ dtypes将分别由每一列的内容确定”。

---

编辑：如@askewchan所建议，您必须传递names=True来保留csv列的原始名称。