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)
