Skip to content

Commit d69fa3b

Browse files
committed
Fix rhs dict assignment
1 parent 85bab01 commit d69fa3b

File tree

2 files changed

+12
-3
lines changed

2 files changed

+12
-3
lines changed

pandas/core/indexing.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,6 @@ def _has_valid_positional_setitem_indexer(self, indexer):
200200
return True
201201

202202
def _setitem_with_indexer(self, indexer, value):
203-
204203
self._has_valid_setitem_indexer(indexer)
205204

206205
# also has the side effect of consolidating in-place
@@ -486,8 +485,8 @@ def can_do_equal_len():
486485
self.obj[item_labels[indexer[info_axis]]] = value
487486
return
488487

489-
if isinstance(value, ABCSeries):
490-
value = self._align_series(indexer, value)
488+
if isinstance(value, (ABCSeries, dict)):
489+
value = self._align_series(indexer, Series(value))
491490

492491
elif isinstance(value, ABCDataFrame):
493492
value = self._align_frame(indexer, value)

pandas/tests/test_indexing.py

+10
Original file line numberDiff line numberDiff line change
@@ -4411,6 +4411,16 @@ def test_slice_with_zero_step_raises(self):
44114411
self.assertRaisesRegexp(ValueError, 'slice step cannot be zero',
44124412
lambda: s.ix[::0])
44134413

4414+
def test_indexing_assignment_dict_already_exists(self):
4415+
df = pd.DataFrame({'x': [1, 2, 6],
4416+
'y': [2, 2, 8],
4417+
'z': [-5, 0, 5]}).set_index('z')
4418+
expected = df.copy()
4419+
rhs = dict(x=9, y=99)
4420+
df.loc[5] = rhs
4421+
expected.loc[5] = [9, 99]
4422+
tm.assert_frame_equal(df, expected)
4423+
44144424

44154425
class TestSeriesNoneCoercion(tm.TestCase):
44164426
EXPECTED_RESULTS = [

0 commit comments

Comments
 (0)