Numpy genfromtxt 中的列名

在本文中，我们将介绍 Numpy 中的 genfromtxt 函数中如何使用和获取列名。

阅读更多：Numpy 教程

genfromtxt 函数简介

先来简单介绍一下 genfromtxt 函数。genfromtxt 是 Numpy 中用于读取文本文件的函数之一，我们可以通过它读取 CSV，TXT 等格式的文件，并将数据存储在一个 NumPy 数组中。genfromtxt 函数通常会返回一个 numpy.ndarray 类型的对象。下面是一个使用 genfromtxt 函数读取 CSV 文件并返回 NumPy 数组的简单示例：

import numpy as np

data = np.genfromtxt('data.csv', delimiter=',')

设置列名

默认情况下，genfromtxt 函数会将第一行数据作为数据而不是列名。如果你的 CSV 文件包含了列名，那么我们可以通过一些参数来指定这些列名。

genfromtxt 函数的 names 参数可以接受多个不同的值来指定列名。下面是设置列名的示例：

data = np.genfromtxt('data.csv', delimiter=',', names=True, dtype=None)

print(data['id'])

这里我们给 names 参数传递了一个布尔值 True，告诉 genfromtxt 函数我们要使用第一行数据作为列名。同时，我们还需要给 dtype 参数传递 None 来使用每一列最适合的数据类型。

这时我们就可以通过获取数组列名来对数据进行处理。上面的示例中 data['id'] 就是获取 NumPy 数组中名为 “id” 的这一列数据。

指定列名

如果你的 CSV 文件不包含列名，那么我们也可以通过 names 参数来指定列名。

data = np.genfromtxt('data.csv', delimiter=',', names=('id', 'name', 'age'), dtype=None)

print(data['name'])

这里，我们给 names 参数传递了一个元组，其中包含了所有的列名。这时，我们就可以通过 data['name'] 来获取数组中名为 “name” 的这一列数据。

自定义列名

除了使用携带列名的 CSV 文件或手动指定列名外，我们还可以通过自定义函数来指定列名。这个函数应该返回一个字符串数组，每个字符串元素都对应一个列的名字。

def get_headers(filename):
    with open(filename, 'r') as f:
        headers = f.readline().strip().split(',')
    return headers

data = np.genfromtxt('data.csv', delimiter=',', names=get_headers('data.csv'), dtype=None)

print(data['name'])

这里我们定义了一个名为 get_headers 的函数，它打开 CSV 文件，读取第一行，然后返回一个由列名组成的字符串数组。接下来，我们将这个函数传递给 genfromtxt 函数的 names 参数，以便使用返回的列名来加载数据。

总结

使用 genfromtxt 函数加载数据时，names 参数可以在读取 CSV 文件时实现列名的添加和指定。无论数据集是否有列名，我们都可以通过功能强大的 genfromtxt 函数来加载它们，并使用 names 参数来指定、自定义或获取列名。现在，我们可以方便地使用 Pandas 和其他一些 libraries 来处理 NumPy 数组。