如何去掉pandas数据框中的“未命名:0”列?

2024-04-24 06:49:14 发布

您现在位置:Python中文网/ 问答频道 /正文

有时当我从df中读取csv时,会得到一个不需要的索引,比如名为unnamed:0的列。

file.csv

,A,B,C
0,1,2,3
1,4,5,6
2,7,8,9

读取CSV时使用以下命令:

pd.read_csv('file.csv')

   Unnamed: 0  A  B  C
0           0  1  2  3
1           1  4  5  6
2           2  7  8  9

这太烦人了!有人知道怎么摆脱这个吗?


Tags: csv命令dfreadfilepdunnamed太烦人
3条回答

这是索引列,传递index=False不写出它,请参见docs

示例:

In [37]:
df = pd.DataFrame(np.random.randn(5,3), columns=list('abc'))
pd.read_csv(io.StringIO(df.to_csv()))

Out[37]:
   Unnamed: 0         a         b         c
0           0  0.109066 -1.112704 -0.545209
1           1  0.447114  1.525341  0.317252
2           2  0.507495  0.137863  0.886283
3           3  1.452867  1.888363  1.168101
4           4  0.901371 -0.704805  0.088335

比较:

In [38]:
pd.read_csv(io.StringIO(df.to_csv(index=False)))

Out[38]:
          a         b         c
0  0.109066 -1.112704 -0.545209
1  0.447114  1.525341  0.317252
2  0.507495  0.137863  0.886283
3  1.452867  1.888363  1.168101
4  0.901371 -0.704805  0.088335

您还可以通过传递index_col=0来告诉read_csv第一列是索引列:

In [40]:
pd.read_csv(io.StringIO(df.to_csv()), index_col=0)

Out[40]:
          a         b         c
0  0.109066 -1.112704 -0.545209
1  0.447114  1.525341  0.317252
2  0.507495  0.137863  0.886283
3  1.452867  1.888363  1.168101
4  0.901371 -0.704805  0.088335

另一种可能发生这种情况的情况是,如果您的数据不正确地写入csv,使每一行以逗号结尾。当您试图将数据读入df时,这将在数据末尾留下一个未命名的列Unnamed: x

这个问题很可能是由于CSV和它的RangeIndex(通常没有名称)一起保存的。在保存数据帧时,实际上需要进行修复,但这并不总是一个选项。

避免问题:read_csv使用index_col参数

在IMO中,最简单的解决方案是将未命名的列读为索引。为^{}指定一个index_col=[0]参数,这将读取第一列作为索引。

df = pd.DataFrame('x', index=range(5), columns=list('abc'))
df

   a  b  c
0  x  x  x
1  x  x  x
2  x  x  x
3  x  x  x
4  x  x  x

# Save DataFrame to CSV.
df.to_csv('file.csv')

pd.read_csv('file.csv')

   Unnamed: 0  a  b  c
0           0  x  x  x
1           1  x  x  x
2           2  x  x  x
3           3  x  x  x
4           4  x  x  x

# Now try this again, with the extra argument.
pd.read_csv('file.csv', index_col=[0])

   a  b  c
0  x  x  x
1  x  x  x
2  x  x  x
3  x  x  x
4  x  x  x

Note
You could have avoided this in the first place by using index=False when creating the output CSV, if your DataFrame does not have an index to begin with.

df.to_csv('file.csv', index=False)

But as mentioned above, this isn't always an option.


权宜之计:用str.match过滤

如果您不能修改代码来读/写CSV文件,您可以使用^{}过滤来删除列:

df 

   Unnamed: 0  a  b  c
0           0  x  x  x
1           1  x  x  x
2           2  x  x  x
3           3  x  x  x
4           4  x  x  x

df.columns
# Index(['Unnamed: 0', 'a', 'b', 'c'], dtype='object')

df.columns.str.match('Unnamed')
# array([ True, False, False, False])

df.loc[:, ~df.columns.str.match('Unnamed')]

   a  b  c
0  x  x  x
1  x  x  x
2  x  x  x
3  x  x  x
4  x  x  x

相关问题 更多 >