Пример #1
0
    def test_delitem_and_pop(self):
        expected = self.panel['ItemA']
        result = self.panel.pop('ItemA')
        assert_frame_equal(expected, result)
        self.assert_('ItemA' not in self.panel.items)

        del self.panel['ItemB']
        self.assert_('ItemB' not in self.panel.items)
        self.assertRaises(Exception, self.panel.__delitem__, 'ItemB')

        values = np.empty((3, 3, 3))
        values[0] = 0
        values[1] = 1
        values[2] = 2

        panel = Panel(values, range(3), range(3), range(3))

        # did we delete the right row?

        panelc = panel.copy()
        del panelc[0]
        assert_frame_equal(panelc[1], panel[1])
        assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[1]
        assert_frame_equal(panelc[0], panel[0])
        assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[2]
        assert_frame_equal(panelc[1], panel[1])
        assert_frame_equal(panelc[0], panel[0])
Пример #2
0
    def test_delitem_and_pop(self):
        expected = self.panel['ItemA']
        result = self.panel.pop('ItemA')
        assert_frame_equal(expected, result)
        self.assert_('ItemA' not in self.panel.items)

        del self.panel['ItemB']
        self.assert_('ItemB' not in self.panel.items)
        self.assertRaises(Exception, self.panel.__delitem__, 'ItemB')

        values = np.empty((3, 3, 3))
        values[0] = 0
        values[1] = 1
        values[2] = 2

        panel = Panel(values, range(3), range(3), range(3))

        # did we delete the right row?

        panelc = panel.copy()
        del panelc[0]
        assert_frame_equal(panelc[1], panel[1])
        assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[1]
        assert_frame_equal(panelc[0], panel[0])
        assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[2]
        assert_frame_equal(panelc[1], panel[1])
        assert_frame_equal(panelc[0], panel[0])
Пример #3
0
    def test_delitem_and_pop(self):

        values = np.empty((3, 3, 3))
        values[0] = 0
        values[1] = 1
        values[2] = 2

        panel = Panel(values, lrange(3), lrange(3), lrange(3))

        # did we delete the right row?

        panelc = panel.copy()
        del panelc[0]
        tm.assert_frame_equal(panelc[1], panel[1])
        tm.assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[1]
        tm.assert_frame_equal(panelc[0], panel[0])
        tm.assert_frame_equal(panelc[2], panel[2])

        panelc = panel.copy()
        del panelc[2]
        tm.assert_frame_equal(panelc[1], panel[1])
        tm.assert_frame_equal(panelc[0], panel[0])
Пример #4
0
    def test_ix_frame_align(self):
        from pandas import DataFrame
        df = DataFrame(np.random.randn(2, 10))
        df.sort_index(inplace=True)
        p_orig = Panel(np.random.randn(3, 10, 2))

        p = p_orig.copy()
        p.ix[0, :, :] = df
        out = p.ix[0, :, :].T.reindex(df.index, columns=df.columns)
        assert_frame_equal(out, df)

        p = p_orig.copy()
        p.ix[0] = df
        out = p.ix[0].T.reindex(df.index, columns=df.columns)
        assert_frame_equal(out, df)

        p = p_orig.copy()
        p.ix[0, [0, 1, 3, 5], -2:] = df
        out = p.ix[0, [0, 1, 3, 5], -2:]
        assert_frame_equal(out, df.T.reindex([0, 1, 3, 5], p.minor_axis[-2:]))
Пример #5
0
    def test_ix_align(self):
        from pandas import Series
        b = Series(np.random.randn(10), name=0)
        b.sort_values()
        df_orig = Panel(np.random.randn(3, 10, 2))
        df = df_orig.copy()

        df.loc[0, :, 0] = b
        assert_series_equal(df.loc[0, :, 0].reindex(b.index), b)

        df = df_orig.swapaxes(0, 1)
        df.loc[:, 0, 0] = b
        assert_series_equal(df.loc[:, 0, 0].reindex(b.index), b)

        df = df_orig.swapaxes(1, 2)
        df.loc[0, 0, :] = b
        assert_series_equal(df.loc[0, 0, :].reindex(b.index), b)
Пример #6
0
    def test_ix_align(self):
        from pandas import Series
        b = Series(np.random.randn(10))
        b.sort()
        df_orig = Panel(np.random.randn(3, 10, 2))
        df = df_orig.copy()

        df.ix[0, :, 0] = b
        assert_series_equal(df.ix[0, :, 0].reindex(b.index), b)

        df = df_orig.swapaxes(0, 1)
        df.ix[:, 0, 0] = b
        assert_series_equal(df.ix[:, 0, 0].reindex(b.index), b)

        df = df_orig.swapaxes(1, 2)
        df.ix[0, 0, :] = b
        assert_series_equal(df.ix[0, 0, :].reindex(b.index), b)