| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556 |
- from warnings import catch_warnings, simplefilter
- import pytest
- from pandas.compat import lrange
- from pandas.errors import PerformanceWarning
- from pandas import DataFrame, MultiIndex
- from pandas.util import testing as tm
- @pytest.mark.filterwarnings("ignore:\\n.ix:DeprecationWarning")
- class TestMultiIndexIx(object):
- def test_frame_setitem_ix(self, multiindex_dataframe_random_data):
- frame = multiindex_dataframe_random_data
- frame.loc[('bar', 'two'), 'B'] = 5
- assert frame.loc[('bar', 'two'), 'B'] == 5
- # with integer labels
- df = frame.copy()
- df.columns = lrange(3)
- df.loc[('bar', 'two'), 1] = 7
- assert df.loc[('bar', 'two'), 1] == 7
- with catch_warnings(record=True):
- simplefilter("ignore", DeprecationWarning)
- df = frame.copy()
- df.columns = lrange(3)
- df.ix[('bar', 'two'), 1] = 7
- assert df.loc[('bar', 'two'), 1] == 7
- def test_ix_general(self):
- # ix general issues
- # GH 2817
- data = {'amount': {0: 700, 1: 600, 2: 222, 3: 333, 4: 444},
- 'col': {0: 3.5, 1: 3.5, 2: 4.0, 3: 4.0, 4: 4.0},
- 'year': {0: 2012, 1: 2011, 2: 2012, 3: 2012, 4: 2012}}
- df = DataFrame(data).set_index(keys=['col', 'year'])
- key = 4.0, 2012
- # emits a PerformanceWarning, ok
- with tm.assert_produces_warning(PerformanceWarning):
- tm.assert_frame_equal(df.loc[key], df.iloc[2:])
- # this is ok
- df.sort_index(inplace=True)
- res = df.loc[key]
- # col has float dtype, result should be Float64Index
- index = MultiIndex.from_arrays([[4.] * 3, [2012] * 3],
- names=['col', 'year'])
- expected = DataFrame({'amount': [222, 333, 444]}, index=index)
- tm.assert_frame_equal(res, expected)
|
