Skip to content

Commit 8806e7f

Browse files
author
Chang She
committed
BUG: CSV parser fails with single line, no separator pandas-dev#1553
1 parent 7f01a44 commit 8806e7f

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

pandas/io/parsers.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -576,7 +576,10 @@ def _infer_columns(self):
576576
counts[col] = cur_count + 1
577577
self._clear_buffer()
578578
else:
579-
line = self._next_line()
579+
if len(self.buf) > 0:
580+
line = self.buf[0]
581+
else:
582+
line = self._next_line()
580583

581584
ncols = len(line)
582585
if not names:

pandas/io/tests/test_parsers.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -252,6 +252,10 @@ def func(*date_cols):
252252
d = datetime(1999, 1, 27, 19, 0)
253253
self.assert_(df.index[0] == d)
254254

255+
def test_single_line(self):
256+
df = read_csv(StringIO('1,2'), names=['a', 'b'], sep=None)
257+
assert_frame_equal(DataFrame({'a': [1], 'b': [2]}), df)
258+
255259
def test_multiple_date_cols_with_header(self):
256260
data = """\
257261
ID,date,NominalTime,ActualTime,TDew,TAir,Windspeed,Precip,WindDir

0 commit comments

Comments
 (0)