File tree 1 file changed +18
-17
lines changed
1 file changed +18
-17
lines changed Original file line number Diff line number Diff line change @@ -2424,26 +2424,27 @@ def test_iloc_mi(self):
2424
2424
# GH 13797
2425
2425
# Test if iloc can handle integer locations in MultiIndexed DataFrame
2426
2426
2427
- data = [
2428
- ['str00' , 'str01' ],
2429
- ['str10' , 'str11' ],
2430
- ['str20' , 'srt21' ],
2431
- ['str30' , 'str31' ],
2432
- ['str40' , 'str41' ]
2433
- ]
2434
-
2435
- mi = pd .MultiIndex .from_tuples (
2436
- [('CC' ,'A' ),
2437
- ('CC' ,'B' ),
2438
- ('CC' ,'B' ),
2439
- ('BB' ,'a' ),
2440
- ('BB' ,'b' )
2441
- ])
2427
+ data = [
2428
+ ['str00' , 'str01' ],
2429
+ ['str10' , 'str11' ],
2430
+ ['str20' , 'srt21' ],
2431
+ ['str30' , 'str31' ],
2432
+ ['str40' , 'str41' ]
2433
+ ]
2434
+
2435
+ mi = pd .MultiIndex .from_tuples (
2436
+ [('CC' , 'A' ),
2437
+ ('CC' , 'B' ),
2438
+ ('CC' , 'B' ),
2439
+ ('BB' , 'a' ),
2440
+ ('BB' , 'b' )
2441
+ ])
2442
2442
2443
2443
ans = pd .DataFrame (data )
2444
- df_mi = pd .DataFrame (data , index = mi )
2444
+ df_mi = pd .DataFrame (data , index = mi )
2445
2445
2446
- res = pd .DataFrame ([[df_mi .iloc [r ,c ] for c in range (2 )] for r in range (5 )])
2446
+ res = pd .DataFrame ([[df_mi .iloc [r , c ] for c in range (2 )]
2447
+ for r in range (5 )])
2447
2448
2448
2449
assert_frame_equal (res , ans )
2449
2450
You can’t perform that action at this time.
0 commit comments