def test_add_row(self): frame = pd.DataFrame({ "department": { "tom": "IT", "mary": "HR" }, "age": { "tom": 25, "mary": 21 } }) frame.ix["alice"] = pd.Series([24, "Law"], index=["age", "department"]) expected = pd.DataFrame({ "department": { "tom": "IT", "mary": "HR", "alice": "Law" }, "age": { "tom": 25, "mary": 21, "alice": 24 } }) tt.ignore_order_assert_frames_equal(expected, frame)
def test_ix1(self): frame = pd.DataFrame(np.arange(1,10).reshape(3,3), columns = ["a","b","c"], index = ["record1","record2","record3"]) subframe = frame.ix[["record3","record1"],["c","a"]] expected = pd.DataFrame({"a":{"record1":1,"record3":7}, "c":{"record1":3,"record3":9}}) tt.ignore_order_assert_frames_equal(expected,subframe)
def test_add_row(self): frame = pd.DataFrame({"department":{"tom":"IT","mary":"HR"}, "age":{"tom":25,"mary":21}}) frame.ix["alice"] = pd.Series([24,"Law"],index=["age","department"]) expected = pd.DataFrame({"department":{"tom":"IT","mary":"HR","alice":"Law"}, "age":{"tom":25,"mary":21,"alice":24}}) tt.ignore_order_assert_frames_equal(expected,frame)
def test_ix1(self): frame = pd.DataFrame(np.arange(1, 10).reshape(3, 3), columns=["a", "b", "c"], index=["record1", "record2", "record3"]) subframe = frame.ix[["record3", "record1"], ["c", "a"]] expected = pd.DataFrame({ "a": { "record1": 1, "record3": 7 }, "c": { "record1": 3, "record3": 9 } }) tt.ignore_order_assert_frames_equal(expected, subframe)