Pandas 将 "\r" 添加到 csv 文件

don*_*234 4 python csv python-3.x pandas

这可以归结为一个简单的问题在这里

我有一个看起来像这样的熊猫数据框:

In [1]: df
Out[1]:
   0        1
0  a  A\nB\nC
1  a  D\nE\nF
2  b  A\nB\nC
Run Code Online (Sandbox Code Playgroud)

当我将其写入 csv 文件然后读回时,我希望具有相同的数据帧。不是这种情况:

In [2]: df.to_csv("out.csv")

In [3]: df = pd.read_csv("out.csv", index_col=0)

In [4]: df
Out[4]:
   0            1
0  a  A\r\nB\r\nC
1  a  D\r\nE\r\nF
2  b  A\r\nB\r\nC
Run Code Online (Sandbox Code Playgroud)

\r在每个 之前添加一个字符\n。再写读一遍,同样的事情发生了:

In [5]: df.to_csv("out.csv")

In [6]: df = pd.read_csv("out.csv", index_col=0)

In [7]: df
Out[7]:
   0                1
0  a  A\r\r\nB\r\r\nC
1  a  D\r\r\nE\r\r\nF
2  b  A\r\r\nB\r\r\nC
Run Code Online (Sandbox Code Playgroud)

如何阻止熊猫添加\r角色?


编辑:
是的,我在 Windows 上。


pd.read_csv(pd.compat.StringIO(df.to_csv(index=False))) 给了我相同的数据帧,所以问题似乎是写入文件
以二进制模式传递一个打开的文件对象,如下所示:

with open("out.csv", "wb") as file:
    df.to_csv(file)
Run Code Online (Sandbox Code Playgroud)

结果是:

TypeError                                 Traceback (most recent call last)
<ipython-input-20-f31d52fb2ce3> in <module>()
      1 with open("out.csv", "wb") as file:
----> 2     df.to_csv(file)
      3

C:\Program Files\Anaconda3\lib\site-packages\pandas\core\frame.py in to_csv(self, path_or_buf, sep, na_rep, float_format, columns, header, index, index_label, mode, encoding, compression, quoting, quotechar, line_terminator, chunksize, tupleize_cols, date_format, doublequote, escapechar, decimal, **kwds)
   1342                                      doublequote=doublequote,
   1343                                      escapechar=escapechar, decimal=decimal)
-> 1344         formatter.save()
   1345
   1346         if path_or_buf is None:

C:\Program Files\Anaconda3\lib\site-packages\pandas\formats\format.py in save(self)
   1549
   1550             else:
-> 1551                 self._save()
   1552
   1553         finally:

C:\Program Files\Anaconda3\lib\site-packages\pandas\formats\format.py in _save(self)
   1636     def _save(self):
   1637
-> 1638         self._save_header()
   1639
   1640         nrows = len(self.data_index)

C:\Program Files\Anaconda3\lib\site-packages\pandas\formats\format.py in _save_header(self)
   1632
   1633         # write out the index label line
-> 1634         writer.writerow(encoded_labels)
   1635
   1636     def _save(self):

TypeError: a bytes-like object is required, not 'str'
Run Code Online (Sandbox Code Playgroud)


使用常规写入无济于事

In [1]: with open("out.csv", "w") as file:
   ...:     df.to_csv(file)
   ...:

In [2]: df = pd.read_csv("out.csv")

In [3]: df
Out[3]:
   Unnamed: 0  0            1
0           0  a  A\r\nB\r\nC
1           1  a  D\r\nE\r\nF
2           2  b  A\r\nB\r\nC
Run Code Online (Sandbox Code Playgroud)


我的python版本是 Python 3.5.2 :: Anaconda 4.2.0 (64-bit)
我已经确定问题出在pandas.read_csv而不是pandas.to_csv

In [1]: df
Out[1]:
   0        1
0  a  A\nB\nC
1  a  D\nE\nF
2  b  A\nB\nC

In [2]: df.to_csv("out.csv")

In [3]: with open("out.csv", "r") as file:
    ...:     s = file.read()
    ...:

In [4]: s  # Only to_csv has been used, no \r's!
Out[4]: ',0,1\n0,a,"A\nB\nC"\n1,a,"D\nE\nF"\n2,b,"A\nB\nC"\n'

In [5]: pd.read_csv("out.csv")  # Now the \r's come in
Out[5]:
   Unnamed: 0  0            1
0           0  a  A\r\nB\r\nC
1           1  a  D\r\nE\r\nF
2           2  b  A\r\nB\r\nC
Run Code Online (Sandbox Code Playgroud)

小智 7

正如一些人在上面的评论和您在此处引用的帖子中所说的那样 ,这是序列化换行符时的典型 Windows 问题。该问题也已在 pandas-dev github #17365上报告。

希望在 Python3 上,您可以指定换行符:

with open("out.csv", mode='w', newline='\n') as f:
    df.to_csv(f, sep=",", line_terminator='\n', encoding='utf-8')
Run Code Online (Sandbox Code Playgroud)