@@ -810,41 +810,42 @@ def test_loc_setitem_with_scalar_index(self, indexer, value):
810810
811811 def test_loc_setitem_missing_columns_scalar_index_list_value (self ):
812812 # GH 26534
813- df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ['A' , 'B' ])
814- df .loc [1 , ['C' , 'D' ]] = [7 , 8 ]
815- expected = pd .DataFrame ([[ 1 , 2 , np . nan , np . nan ],
816- [3 , 4 , 7 , 8 ],
817- [ 5 , 6 , np . nan , np . nan ] ],
818- columns = [ 'A' , 'B' , 'C' , 'D' ] )
813+ df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ["A" , "B" ])
814+ df .loc [1 , ["C" , "D" ]] = [7 , 8 ]
815+ expected = pd .DataFrame (
816+ [[ 1 , 2 , np . nan , np . nan ], [3 , 4 , 7 , 8 ], [ 5 , 6 , np . nan , np . nan ] ],
817+ columns = [ "A" , "B" , "C" , "D" ],
818+ )
819819 tm .assert_frame_equal (df , expected )
820820
821821 def test_loc_setitem_missing_columns_full_index_dataframe_value (self ):
822822 # GH 26534
823- df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ['A' , 'B' ])
824- df2 = pd .DataFrame ([[7 , 8 ], [9 , 10 ], [11 , 12 ]], columns = ['A' , 'C' ])
825- df .loc [:, ['A' , 'C' ]] = df2
826- expected = pd .DataFrame ([[7 , 2 , 8 ], [9 , 4 , 10 ], [11 , 6 , 12 ]],
827- columns = ['A' , 'B' , 'C' ])
823+ df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ["A" , "B" ])
824+ df2 = pd .DataFrame ([[7 , 8 ], [9 , 10 ], [11 , 12 ]], columns = ["A" , "C" ])
825+ df .loc [:, ["A" , "C" ]] = df2
826+ expected = pd .DataFrame (
827+ [[7 , 2 , 8 ], [9 , 4 , 10 ], [11 , 6 , 12 ]], columns = ["A" , "B" , "C" ]
828+ )
828829 tm .assert_frame_equal (df , expected )
829830
830831 def test_loc_setitem_missing_columns_list_index_scalar_value (self ):
831832 # GH 26534
832- df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ['A' , 'B' ])
833- df .loc [[0 , 2 ], ['B' , 'C' , 'D' ]] = 9
834- expected = pd .DataFrame ([[ 1 , 9 , 9 , 9 ],
835- [3 , 4 , np .nan , np .nan ],
836- [ 5 , 9 , 9 , 9 ] ],
837- columns = [ 'A' , 'B' , 'C' , 'D' ] )
833+ df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ["A" , "B" ])
834+ df .loc [[0 , 2 ], ["B" , "C" , "D" ]] = 9
835+ expected = pd .DataFrame (
836+ [[ 1 , 9 , 9 , 9 ], [3 , 4 , np .nan , np .nan ], [ 5 , 9 , 9 , 9 ] ],
837+ columns = [ "A" , "B" , "C" , "D" ],
838+ )
838839 tm .assert_frame_equal (df , expected )
839840
840841 def test_loc_setitem_missing_columns_range_index_2dlist_value (self ):
841842 # GH 26534
842- df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ['A' , 'B' ])
843- df .loc [1 :3 , ['B' , 'C' , 'D' ]] = [[7 , 8 , 9 ], [10 , 11 , 12 ]]
844- expected = pd .DataFrame ([[ 1 , 2 , np . nan , np . nan ],
845- [3 , 7 , 8 , 9 ],
846- [ 5 , 10 , 11 , 12 ] ],
847- columns = [ 'A' , 'B' , 'C' , 'D' ] )
843+ df = pd .DataFrame ([[1 , 2 ], [3 , 4 ], [5 , 6 ]], columns = ["A" , "B" ])
844+ df .loc [1 :3 , ["B" , "C" , "D" ]] = [[7 , 8 , 9 ], [10 , 11 , 12 ]]
845+ expected = pd .DataFrame (
846+ [[ 1 , 2 , np . nan , np . nan ], [3 , 7 , 8 , 9 ], [ 5 , 10 , 11 , 12 ] ],
847+ columns = [ "A" , "B" , "C" , "D" ],
848+ )
848849 tm .assert_frame_equal (df , expected )
849850
850851 def test_loc_coercion (self ):
0 commit comments