Skip to content

Commit 86da483

Browse files
author
fengyurun
committed
fix index name
1 parent efe6a90 commit 86da483

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

databricks/koalas/indexes.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ def fillna(self, value):
387387
if not isinstance(value, (float, int, str, bool)):
388388
raise TypeError("Unsupported type %s" % type(value))
389389
sdf = self._internal.sdf.fillna(value)
390-
result = Index(DataFrame(_InternalFrame(sdf=sdf, index_map=self._internal.index_map)))
390+
result = DataFrame(self._kdf._internal.copy(sdf=sdf)).index
391391
return result
392392

393393
def to_series(self, name: Union[str, Tuple[str, ...]] = None) -> Series:

databricks/koalas/tests/test_indexes.py

+1
Original file line numberDiff line numberDiff line change
@@ -325,6 +325,7 @@ def test_index_fillna(self):
325325
kidx = ks.DataFrame({'a': ['a', 'b', 'c']}, index=[1, 2, None]).index
326326

327327
self.assert_eq(pidx.fillna(0), kidx.fillna(0))
328+
self.assert_eq(pidx.rename('name').fillna(0), kidx.rename('name').fillna(0))
328329

329330
with self.assertRaisesRegex(TypeError, "Unsupported type <class 'list'>"):
330331
kidx.fillna([1, 2])

0 commit comments

Comments
 (0)