首页 > 解决方案 > 如何在numpy中获取特定列的列标题

问题描述

我有以下数据

admit_data = np.genfromtxt('/content/drive/My Drive/Colab/admission_predict.csv', delimiter=',')

我需要的是获得一些特定的列标题。我正在使用以下代码来获取数据。但无法获得那些列名

print(admit_data[1:].tolist())

有什么功能.tolist()可以让我只提取该列的名称吗?

编辑 1

添加了示例数据格式

在此处输入图像描述

标签: pythonnumpy

解决方案


您能否提供有关您要提取的数据的更多信息。

根据您的问题, tolist() 函数适用于 Pandas 系列。更好地将 admission_data 转换为 pandas 系列(使用 pd.Series() 函数)。然后您可以将第一行提取为列表。


推荐阅读