时间戳

首页 > 解决方案 > 使用 pandas 从 csv 文件中读取特征名称

python - 使用 pandas 从 csv 文件中读取特征名称

问题描述

我有一个 csv 文件看起来像

        F1    F2    F3
 A1      2     4     2
 A2      4     1     2

当我使用 pandas 读取文件时，我看到第一列未命名。

import pandas as pd
df = pd.read_csv("data.csv")
features = df.columns
print( features )

Index(['Unnamed: 0', 'F1, 'F2, 'F3'])

事实上，我只想获得 F1、F2 和 F3。我可以通过一些数组操作来解决这个问题。但我想知道 pandas 是否有一些内置功能可以做到这一点。任何想法？

更新：

使用index_col = FalseorNone也不起作用。

在此处输入图像描述

标签： pythonpandas

解决方案

那个未命名只是因为索引列被读取，可以使用index_col = [0]read语句中的参数来解析。

这会选择第一列作为索引而不是特征本身。

import pandas as pd
df = pd.read_csv("data.csv", index_col=[0])
features = df.columns
print( features )

Index(['F1', 'F2', 'F3'])

推荐阅读