主页 / 问题 / 1036067
Accepted
Xander
Asked:2020-10-18 20:32:53 +0000 UTC2020-10-18 20:32:53 +0000 UTC

如果数据列少于名称，如何读取 CSV，用空值填充它们

  • 772

遇到这样的问题。

csv 中的数据从外部进入我的代码。我从中创建了一个数据框并进一步使用它。参数如下所示：

df = pandas.read_csv(f, header=None, index_col=False,
                     delimiter='\t',
                     names=name_list, dtype=dtype_dict, 
                     converters=converters_dict, usecols=usecols_list
                     )

问题是一段时间以来数据的来源现在给 CSV 没有最后一列的某些数据，在这种情况下，我的代码因错误而崩溃：

Too many columns specified: expected 21 and found 20

在这种情况下，我需要代码来创建一个包含 21 列的数据框，只有最后一个应该用空值填充。

当然，您可以对传入 CSV 中的列数进行初步检查，根据这一点，创建一个名称为 20 或 21 个值的列表，然后添加或不添加另一列。

但它有点像拐杖。如果数据中的名称多于实际列，是否有让熊猫自己创建最后一个空列的设置？

python

2 个回答

  1. Best Answer

    除非您确定数据量或确定 和 的长度相等，否则您可能应该使用该参数names而不使用该参数。usecolsnamesusecols

    源csv：

    a,b
a,b,c,d
a,b,c
a,b,c

    编码：

    import pandas as pd

df = pd.read_csv('111.csv', names=list(range(1,10)))
print(df)

    结果：

       1  2    3    4   5   6   7   8   9
0  a  b  NaN  NaN NaN NaN NaN NaN NaN
1  a  b    c    d NaN NaN NaN NaN NaN
2  a  b    c  NaN NaN NaN NaN NaN NaN
3  a  b    c  NaN NaN NaN NaN NaN NaN
    • 2

  2. 我不知道是否有任何单独的参数，想到的唯一决定是通过append()添加从 CSV 文件创建的数据框来创建一个具有所需列数的空数据框。

    In [6]: import pandas as pd

In [7]: pd.read_csv('test.csv')
Out[7]: 
   one  two
0    1    2
1    2    3
2    5    6

In [8]: df = pd.DataFrame(columns=['one', 'two', 'three'])

In [9]: df
Out[9]: 
Empty DataFrame
Columns: [one, two, three]
Index: []

In [10]: df = df.append(pd.read_csv('test.csv'), sort=False)

In [11]: df
Out[11]: 
  one two three
0   1   2   NaN
1   2   3   NaN
2   5   6   NaN
    • 1

相关问题