python - 将 CSV 文件读取到 numpy 数组，第一行为字符串，其余为 float

Question

我将数据存储在 CSV 中，其中第一行是字符串(列名)，其余行是数字。如何将其存储到 numpy 数组中？我能找到的只是如何为列设置数据类型，而不是为行设置数据类型。

现在我只是跳过标题来进行计算，但我需要在最终版本中包含标题。但是，如果我将标题保留在其中，则会将整个数组设置为字符串，并且计算会失败。

这就是我所拥有的:

 data = np.genfromtxt(path_to_csv, dtype=None, delimiter=',', skip_header=1) 

Answer 1

如果您在函数 np.genfromtxt

中使用 names=True 参数，则可以保留列名

 data = np.genfromtxt(path_to_csv, dtype=float, delimiter=',', names=True) 

请注意 dtype=float，它将您的数据转换为 float 。这比使用 dtype=None 更有效，后者要求 np.genfromtxt 为您猜测数据类型。

输出将是一个结构化数组，您可以在其中按名称访问各个列。这些名字将取自您的第一行。可能会发生一些修改，例如列名中的空格将更改为 _。 documentation应该涵盖您可能遇到的大多数问题。